Update app.py

app.py

@@ -2,25 +2,27 @@ import gradio as gr
 import openai
 import os
 
-# Get OpenAI API key from environment
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# OpenAI API setup
+api_key = os.getenv("OPENAI_API_KEY")
+client = openai.OpenAI(api_key=api_key)
 
 def ai_calculator(num1, num2, operation):
-    if not openai.api_key:
+    if not api_key:
         return "❌ OpenAI API key not set."
 
-    prompt = f"""You are a smart calculator. Perform the following operation:
-Input: {num1}, {num2}, {operation}
-Output:"""
+    prompt = f"Perform {operation} operation on {num1} and {num2}. Just return the result only."
 
     try:
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=prompt,
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are a helpful and accurate calculator."},
+                {"role": "user", "content": prompt}
+            ],
             temperature=0,
-            max_tokens=50,
+            max_tokens=50
         )
-        result = response.choices[0].text.strip()
+        result = response.choices[0].message.content.strip()
         return f"✅ Result: {result}"
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -34,7 +36,7 @@ demo = gr.Interface(
     ],
     outputs="text",
     title="🧠 AI Calculator",
-    description="A simple calculator using OpenAI GPT. Choose numbers and operation to get result.",
+    description="A simple calculator using OpenAI GPT-3.5. Choose numbers and operation to get result.",
 )
 
 if __name__ == "__main__":