Update app.py

app.py

@@ -1,27 +1,48 @@
 # app.py
 import streamlit as st
+import math
 
-st.title("Simple Calculator")
+st.title("Enhanced Calculator")
 
 # Input fields for numbers
 num1 = st.number_input("Enter first number")
 num2 = st.number_input("Enter second number")
+num3 = st.number_input("Enter third number")
 
 # Operation selection
-operation = st.selectbox("Select operation", ["+", "-", "*", "/"])
+operation = st.selectbox(
+    "Select operation",
+    ["+", "-", "*", "/", "Power", "Square Root", "Cube Root", "Average"],
+)
 
 # Calculation and display
 if st.button("Calculate"):
-    if operation == "+":
-        result = num1 + num2
-    elif operation == "-":
-        result = num1 - num2
-    elif operation == "*":
-        result = num1 * num2
-    elif operation == "/":
-        if num2 == 0:
-            st.error("Division by zero is not allowed.")
-            st.stop()  # Stop execution if division by zero
-        else:
-            result = num1 / num2
-    st.write(f"Result: {result}")
+    try:  # Wrap in a try-except block for better error handling
+        if operation == "+":
+            result = num1 + num2 + num3
+        elif operation == "-":
+            result = num1 - num2 - num3
+        elif operation == "*":
+            result = num1 * num2 * num3
+        elif operation == "/":
+            if num2 == 0 or num3 == 0:
+                raise ZeroDivisionError("Division by zero is not allowed.")  # Raise exception
+            else:
+                result = num1 / num2 / num3
+        elif operation == "Power":
+            result = num1**num2
+        elif operation == "Square Root":
+            if num1 < 0:
+                raise ValueError("Square root of a negative number is not allowed.")  # Raise exception
+            else:
+                result = math.sqrt(num1)
+        elif operation == "Cube Root":
+            result = num1**(1/3)
+        elif operation == "Average":
+            result = (num1 + num2 + num3) / 3
+        st.write(f"Result: {result}")
+
+    except (ZeroDivisionError, ValueError) as e:  # Catch specific exceptions
+        st.error(str(e))  # Display the error message to the user
+    except Exception as e:  # Catch any other unexpected exceptions
+        st.error(f"An error occurred: {e}")  # Display a generic error message