Update app.py

app.py

@@ -1,43 +1,37 @@
 import openai
-import re
 import gradio as gr
 
-# Define a function to solve math problems
-def solve_math_problem(prompt, api_key):
-    # Set up the OpenAI API credentials
-    openai.api_key = api_key
-
-    # Send the prompt to OpenAI's GPT-3 engine
-    response = openai.Completion.create(
-        engine="davinci",
-        prompt=prompt,
-        max_tokens=100,
-        n=1,
-        stop=None,
-        temperature=0.5,
-    )
+def chat(api_key, message, model):
+    if not api_key:
+        return "Please enter a valid API key."
 
-    # Extract the answer from the response
-    answer = response.choices[0].text.strip()
+    openai.api_key = api_key
 
-    # Remove any non-numeric characters from the answer
-    answer = re.sub("[^0-9\.]", "", answer)
+    try:
+        response = openai.Completion.create(
+            engine=model,
+            prompt=message,
+            max_tokens=50,
+            n=1,
+            stop=None,
+            temperature=0.5,
+        )
+        return response.choices[0].text.strip()
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    return float(answer)
+models = ["gpt-4", "text-davinci-002", "text-curie-002", "text-babbage-002", "text-ada-002"]
 
-# Define the Gradio interface
 iface = gr.Interface(
-    fn=solve_math_problem,
+    fn=chat,
     inputs=[
-        gr.inputs.Textbox(label="Enter a math problem"),
-        gr.inputs.Textbox(label="Enter your OpenAI API key", type="password")
+        gr.inputs.Textbox(lines=1, label="API Key"),
+        gr.inputs.Textbox(lines=5, label="Message"),
+        gr.inputs.Dropdown(choices=models, label="Model"),
     ],
-    outputs=gr.outputs.Textbox(label="Answer"),
-    title="OpenAI Calculator",
-    description="Enter a math problem and let OpenAI solve it for you!",
-    live=True,
-    button_text="Calculate"
+    outputs=gr.outputs.Textbox(label="Response"),
+    title="GPT-4 Chat App",
+    description="A simple chat app using OpenAI GPT-4 and Gradio.",
 )
 
-# Launch the Gradio interface
-iface.launch()
+iface.launch()