Update app.py

app.py

@@ -3,15 +3,13 @@ import math
 
 st.set_page_config(page_title="Calculator", layout="centered")
 
-# ---------------- CSS (SAFE) ---------------- #
+# ---------- Minimal SAFE CSS ---------- #
 st.markdown("""
 <style>
 .calc {
     max-width: 360px;
     margin: auto;
 }
-
-/* Display */
 .display {
     background: #111;
     color: white;
@@ -22,102 +20,65 @@ st.markdown("""
     margin-bottom: 14px;
     min-height: 60px;
 }
-
-/* ALL buttons */
-.stButton > button {
+.stButton button {
     height: 64px;
     font-size: 22px;
     border-radius: 16px;
-    border: none;
-}
-
-/* Operator buttons */
-.stButton > button[data-op="op"] {
-    background-color: #ff9500;
-    color: white;
-}
-
-/* Equal button */
-.stButton > button[data-op="eq"] {
-    background-color: #34c759;
-    color: white;
-}
-
-/* Clear button */
-.stButton > button[data-op="clr"] {
-    background-color: #ff3b30;
-    color: white;
 }
 </style>
 """, unsafe_allow_html=True)
 
-# ---------------- State ---------------- #
+# ---------- State ---------- #
 if "expr" not in st.session_state:
     st.session_state.expr = "0"
 
-def press(val):
-    if val == "C":
+def press(v):
+    if v == "C":
         st.session_state.expr = "0"
-
-    elif val == "=":
+    elif v == "=":
         try:
             st.session_state.expr = str(eval(st.session_state.expr))
         except:
             st.session_state.expr = "Error"
-
-    elif val == "√":
+    elif v == "√":
         try:
             st.session_state.expr = str(math.sqrt(float(st.session_state.expr)))
         except:
             st.session_state.expr = "Error"
-
-    elif val == "x²":
+    elif v == "x²":
         try:
             st.session_state.expr = str(float(st.session_state.expr) ** 2)
         except:
             st.session_state.expr = "Error"
-
-    elif val == "log":
+    elif v == "log":
         try:
             st.session_state.expr = str(math.log10(float(st.session_state.expr)))
         except:
             st.session_state.expr = "Error"
-
     else:
         if st.session_state.expr == "0":
-            st.session_state.expr = val
+            st.session_state.expr = v
         else:
-            st.session_state.expr += val
+            st.session_state.expr += v
 
     st.rerun()
 
-# ---------------- UI ---------------- #
+# ---------- UI ---------- #
 st.markdown('<div class="calc">', unsafe_allow_html=True)
 st.markdown(f'<div class="display">{st.session_state.expr}</div>', unsafe_allow_html=True)
 
-layout = [
-    [("C", "clr"), ("√", "op"), ("x²", "op"), ("/", "op")],
-    [("7", ""), ("8", ""), ("9", ""), ("*", "op")],
-    [("4", ""), ("5", ""), ("6", ""), ("-", "op")],
-    [("1", ""), ("2", ""), ("3", ""), ("+", "op")],
-    [("0", ""), (".", ""), ("log", "op"), ("=", "eq")]
+keys = [
+    ["C", "√", "x²", "/"],
+    ["7", "8", "9", "*"],
+    ["4", "5", "6", "-"],
+    ["1", "2", "3", "+"],
+    ["0", ".", "log", "="]
 ]
 
-for row in layout:
+for row in keys:
     cols = st.columns(4)
-    for i, (label, role) in enumerate(row):
-        with cols[i]:
-            btn = st.button(label, use_container_width=True)
-            st.markdown(
-                f"""
-                <script>
-                const btns = window.parent.document.querySelectorAll("button");
-                btns[btns.length-1].setAttribute("data-op", "{role}");
-                </script>
-                """,
-                unsafe_allow_html=True
-            )
-            if btn:
-                press(label)
+    for i, k in enumerate(row):
+        if cols[i].button(k, use_container_width=True):
+            press(k)
 
 st.markdown("</div>", unsafe_allow_html=True)