Update app.py

app.py

@@ -1,45 +1,40 @@
 import gradio as gr
 
-def calculator(num1, num2, operation):
+def calculator(input_text):
     try:
-        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"
+        return str(eval(input_text))
+    except:
+        return "Error"
 
-# Creating the interface
-with gr.Blocks() as demo:
+def update_input(input_text, new_value):
+    if new_value == "C":
+        return ""
+    elif new_value == "=":
+        try:
+            return str(eval(input_text))
+        except:
+            return "Error"
+    else:
+        return input_text + new_value
+
+with gr.Blocks() as iface:
     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"
-    )
+        input_display = gr.Textbox(value="", label="Display")
     
-    calculate_btn = gr.Button("Calculate")
-    result = gr.Textbox(label="Result")
+    button_layout = [
+        ["7", "8", "9", "/"],
+        ["4", "5", "6", "*"],
+        ["1", "2", "3", "-"],
+        ["0", ".", "C", "+"],
+        ["="]
+    ]
     
-    calculate_btn.click(
-        fn=calculator,
-        inputs=[num1_input, num2_input, operation_dropdown],
-        outputs=result
-    )
+    for row in button_layout:
+        with gr.Row():
+            for button in row:
+                btn = gr.Button(button)
+                btn.click(update_input, inputs=[input_display, btn], outputs=input_display)
 
-if __name__ == "__main__":
-    demo.launch()
+iface.launch()