Update app.py

app.py

@@ -1,23 +1,57 @@
 import gradio as gr
-from calculator.operations import perform_calculation
-from calculator.utils import extract_operation, extract_numbers
+import re
+
+def extract_numbers(text):
+    # Extract numbers from input text
+    return [float(num) for num in re.findall(r"[-+]?\d*\.\d+|\d+", text)]
+
+def extract_operation(text):
+    # Determine the operation based on keywords
+    if "add" in text or "plus" in text or "+" in text:
+        return "add"
+    elif "subtract" in text or "minus" in text or "-" in text:
+        return "subtract"
+    elif "multiply" in text or "times" in text or "*" in text:
+        return "multiply"
+    elif "divide" in text or "divided by" in text or "/" in text:
+        return "divide"
+    else:
+        return None
+
+def perform_calculation(numbers, operation):
+    if len(numbers) < 2:
+        return "Please provide at least two numbers."
+    
+    a, b = numbers[0], numbers[1]
+
+    if operation == "add":
+        return a + b
+    elif operation == "subtract":
+        return a - b
+    elif operation == "multiply":
+        return a * b
+    elif operation == "divide":
+        return a / b if b != 0 else "Cannot divide by zero"
+    else:
+        return "Unsupported operation."
 
 def calculation_interface(question):
-    operation = extract_operation(question)
+    operation = extract_operation(question.lower())
     numbers = extract_numbers(question)
     if not operation or not numbers:
-        return "Invalid input! Please provide a valid mathematical question."
+        return "❌ Invalid input! Please provide a valid mathematical question."
     result = perform_calculation(numbers, operation)
-    return f"The result of the operation is: {result}"
+    return f"✅ Result: {result}"
 
 # Gradio interface setup
 iface = gr.Interface(
-    fn=calculation_interface, 
-    inputs="text", 
-    outputs="text", 
-    title="Simple Calculation App", 
-    description="Provide a mathematical question, and the app will perform the operation."
+    fn=calculation_interface,
+    inputs="text",
+    outputs="text",
+    title="🧮 Simple Calculation App",
+    description="Type something like 'Add 5 and 3' or 'What is 10 divided by 2?'"
 )
 
 if __name__ == "__main__":
     iface.launch()
+