app.py

import gradio as gr
from statistics import mean
from torch.utils.data import Dataset
from collections import OrderedDict
import xml.etree.ElementTree as ET
import openai # For GPT-3 API ...
import os
import multiprocessing
import json
import numpy as np
import random
import torch
import torchtext
import re
import random
import time
import datetime
import pandas as pd
import sys

openai.api_key = os.getenv("api_key")

def greet(question="Xiaoming have 3 apples and he eats 1. After that he share half of them to Mei. How many apples he left?"):
    input = question + '\n\n' + "|step|subquestion|process|result|"
    response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo",
                messages=[
                        {"role": "system", "content": "You are a helpful assistant that generate table to solve reasoning problem."},
                        {"role": "user", "content": input},

                    ]
                )
    response = response["choices"][0]["message"]["content"]
    return "|step|subquestion|process|result|\n" + response


iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()