Update app.py

app.py

@@ -1,22 +1,39 @@
 # app.py
+
 import gradio as gr
 
-# Define the calculator function
-def calculator(expression):
+# Define the calculator logic
+def simple_calculator(num1, num2, operation):
     try:
-        result = eval(expression)
-        return result
-    except:
-        return "Invalid input"
+        if operation == "+":
+            result = num1 + num2
+        elif operation == "-":
+            result = num1 - num2
+        elif operation == "*":
+            result = num1 * num2
+        elif operation == "/":
+            result = num1 / num2 if num2 != 0 else "∞ (Divide by Zero)"
+        else:
+            result = "Unknown operation"
+        return f"✅ Result: {result}"
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
 
 # Create Gradio interface
-iface = gr.Interface(
-    fn=calculator,
-    inputs=gr.Textbox(label="Enter a math expression (e.g., 5 + 3 * 2)"),
+demo = gr.Interface(
+    fn=simple_calculator,
+    inputs=[
+        gr.Number(label="Enter First Number"),
+        gr.Number(label="Enter Second Number"),
+        gr.Radio(["+", "-", "*", "/"], label="Select Operation"),
+    ],
     outputs=gr.Textbox(label="Result"),
-    title="🧮 Calculator",
-    description="This calculator can do addition (+), subtraction (-), multiplication (*), and division (/)."
+    title="🧮 AI Calculator",
+    description="Perform basic math operations: Addition, Subtraction, Multiplication, and Division.",
+    allow_flagging="never",
+    theme="default"
 )
 
-# Launch the app
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()
+