Update app.py

app.py

@@ -1,23 +1,45 @@
 import gradio as gr
 
-def calculator(input_str):
+def calculator(num1, num2, operation):
     try:
-        result = eval(input_str)
-        return result
-    except ZeroDivisionError:
-        return "Division by zero error"
-    except Exception as e:
-        return f"Error: {str(e)}"
+        num1, num2 = float(num1), float(num2)
+        if operation == "Add":
+            return num1 + num2
+        elif operation == "Subtract":
+            return num1 - num2
+        elif operation == "Multiply":
+            return num1 * num2
+        elif operation == "Divide":
+            if num2 == 0:
+                return "Error: Division by zero!"
+            return num1 / num2
+        else:
+            return "Invalid operation"
+    except ValueError:
+        return "Please enter valid numbers"
 
-# Create the Gradio interface
-interface = gr.Interface(
-    fn=calculator,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter calculation (e.g., 3 + 5 * 2)", label="Calculator Input"),
-    outputs="text",
-    title="Simple Calculator",
-    description="Enter a mathematical expression to calculate the result."
-)
+# Creating the interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Simple Calculator")
+    
+    with gr.Row():
+        num1_input = gr.Number(label="First Number")
+        num2_input = gr.Number(label="Second Number")
+    
+    operation_dropdown = gr.Dropdown(
+        choices=["Add", "Subtract", "Multiply", "Divide"],
+        label="Operation",
+        value="Add"
+    )
+    
+    calculate_btn = gr.Button("Calculate")
+    result = gr.Textbox(label="Result")
+    
+    calculate_btn.click(
+        fn=calculator,
+        inputs=[num1_input, num2_input, operation_dropdown],
+        outputs=result
+    )
 
-# Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()