Update app.py

app.py

@@ -1,39 +1,41 @@
-import streamlit as st
+import gradio as gr
 import openai
 import os
 
-# Set API key from environment variable
-api_key = os.getenv("OPENAI_API_KEY")
-
-st.set_page_config(page_title="AI Calculator", page_icon="🧮")
-
-st.title("🧠 AI Math Calculator")
-st.write("Enter two numbers and choose an operation:")
-
-if not api_key:
-    st.error("Please add your OpenAI API key in the Hugging Face secrets as OPENAI_API_KEY")
-else:
-    openai.api_key = api_key
-
-    num1 = st.number_input("Enter first number", format="%.2f")
-    num2 = st.number_input("Enter second number", format="%.2f")
-    operation = st.selectbox("Choose operation", ["add", "subtract", "multiply", "divide"])
-
-    if st.button("Calculate"):
-        prompt = f"""You are a smart calculator. Perform the operation between two numbers and return the result.
-
-        Input: {num1}, {num2}, {operation}
-        Output:"""
-
-        try:
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=prompt,
-                temperature=0,
-                max_tokens=50
-            )
-
-            result = response.choices[0].text.strip()
-            st.success(f"Result: {result}")
-        except Exception as e:
-            st.error(f"Error from OpenAI: {str(e)}")
+# Get OpenAI API key from environment
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+def ai_calculator(num1, num2, operation):
+    if not openai.api_key:
+        return "❌ OpenAI API key not set."
+
+    prompt = f"""You are a smart calculator. Perform the following operation:
+Input: {num1}, {num2}, {operation}
+Output:"""
+
+    try:
+        response = openai.Completion.create(
+            engine="text-davinci-003",
+            prompt=prompt,
+            temperature=0,
+            max_tokens=50,
+        )
+        result = response.choices[0].text.strip()
+        return f"✅ Result: {result}"
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+demo = gr.Interface(
+    fn=ai_calculator,
+    inputs=[
+        gr.Number(label="Enter First Number"),
+        gr.Number(label="Enter Second Number"),
+        gr.Radio(["add", "subtract", "multiply", "divide"], label="Operation"),
+    ],
+    outputs="text",
+    title="🧠 AI Calculator",
+    description="A simple calculator using OpenAI GPT. Choose numbers and operation to get result.",
+)
+
+if __name__ == "__main__":
+    demo.launch()