Update app.py

app.py

@@ -2,8 +2,8 @@ import openai
 import re
 import gradio as gr
 
-# Define a function to generate OpenAI responses
-def generate_response(prompt, api_key):
+# 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
 
@@ -11,29 +11,32 @@ def generate_response(prompt, api_key):
     response = openai.Completion.create(
         engine="davinci",
         prompt=prompt,
-        max_tokens=150,
+        max_tokens=100,
         n=1,
         stop=None,
-        temperature=0.7,
+        temperature=0.5,
     )
 
-    # Extract the response from the API result
+    # Extract the answer from the response
     answer = response.choices[0].text.strip()
 
-    return answer
+    # Remove any non-numeric characters from the answer
+    answer = re.sub("[^0-9\.]", "", answer)
+
+    return float(answer)
 
 # Define the Gradio interface
 iface = gr.Interface(
-    fn=generate_response,
+    fn=solve_math_problem,
     inputs=[
-        gr.inputs.Textbox(label="Enter your message"),
+        gr.inputs.Textbox(label="Enter a math problem"),
         gr.inputs.Textbox(label="Enter your OpenAI API key", type="password")
     ],
-    outputs=gr.outputs.Textbox(label="OpenAI's response"),
-    title="OpenAI Chatbot",
-    description="Ask a question or say something, and OpenAI will respond!",
+    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="Chat"
+    button_text="Calculate"
 )
 
 # Launch the Gradio interface