Update app.py

app.py

@@ -1,30 +1,40 @@
-import gradio as gr
 from fractions import Fraction
+import gradio as gr
 
 def calculator(num1, operation, num2=None):
     if operation == "add":
-        result = Fraction(num1) + Fraction(num2)
+        return num1 + num2
     elif operation == "subtract":
-        result = Fraction(num1) - Fraction(num2)
+        return num1 - num2
     elif operation == "multiply":
-        result = Fraction(num1) * Fraction(num2)
+        return num1 * num2
     elif operation == "divide":
-        result = Fraction(num1) / Fraction(num2)
+        return num1 / num2
     elif operation == "square":
-        result = Fraction(num1) * Fraction(num1)
+        return num1 * num1
     elif operation == "cube":
-        result = Fraction(num1) * Fraction(num1) * Fraction(num1)
+        return num1 * num1 * num1
     elif operation == "exponential":
-        result = pow(Fraction(num1), Fraction(num2))
+        return num1 ** num2
     elif operation == "frac add":
-        result = Fraction(num1) + Fraction(num2)
+        return num1 + num2
     elif operation == "frac subtract":
-        result = Fraction(num1) - Fraction(num2)
+        return num1 - num2
     elif operation == "frac multiply":
-        result = Fraction(num1) * Fraction(num2)
+        return num1 * num2
     elif operation == "frac divide":
-        result = Fraction(num1) / Fraction(num2)
-    return result
+        return num1 / num2
+
+    # convert all inputs to fractions
+    num1 = Fraction(num1)
+    if num2 is not None:
+        num2 = Fraction(num2)
+
+    # convert result back to float if input was not a fraction
+    if isinstance(num1, float) or isinstance(num2, float):
+        return float(calculator(float(num1), operation, float(num2)))
+    else:
+        return num1
 
 demo = gr.Interface(
     fn=calculator,
@@ -40,10 +50,10 @@ demo = gr.Interface(
         [-4, "multiply", 2.5],
         [0, "subtract", 1.2],
         [2, "exponential", 3],
-        [Fraction(3, 4), "frac add", Fraction(1, 4)],
-        [Fraction(3, 4), "frac subtract", Fraction(1, 4)],
-        [Fraction(3, 4), "frac multiply", Fraction(1, 4)],
-        [Fraction(3, 4), "frac divide", Fraction(1, 4)],
+        [Fraction(1, 2), "frac add", Fraction(1, 3)],
+        [Fraction(3, 4), "frac subtract", Fraction(1, 5)],
+        [Fraction(2, 3), "frac multiply", Fraction(4, 5)],
+        [Fraction(3, 4), "frac divide", Fraction(1, 2)]
     ],
     title="Scientific Calculator",
     description="Here's a sample scientific calculator. Enjoy! Code by: Freddy Aboulton   Improved by: Usually3 and ChatGpt"