Update app.py

app.py

@@ -3,71 +3,56 @@ import math
 
 st.set_page_config(page_title="Calculator", layout="centered")
 
-# ---------- Minimal SAFE CSS ---------- #
-st.markdown("""
-<style>
-.calc {
-    max-width: 360px;
-    margin: auto;
-}
-.display {
-    background: #111;
-    color: white;
-    padding: 18px;
-    font-size: 32px;
-    border-radius: 14px;
-    text-align: right;
-    margin-bottom: 14px;
-    min-height: 60px;
-}
-.stButton button {
-    height: 64px;
-    font-size: 22px;
-    border-radius: 16px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-# ---------- State ---------- #
+# ---------- STATE ----------
 if "expr" not in st.session_state:
-    st.session_state.expr = "0"
+    st.session_state.expr = ""
 
-def press(v):
-    if v == "C":
-        st.session_state.expr = "0"
-    elif v == "=":
+def press(key):
+    if key == "C":
+        st.session_state.expr = ""
+    elif key == "=":
         try:
             st.session_state.expr = str(eval(st.session_state.expr))
         except:
             st.session_state.expr = "Error"
-    elif v == "√":
+    elif key == "√":
         try:
             st.session_state.expr = str(math.sqrt(float(st.session_state.expr)))
         except:
             st.session_state.expr = "Error"
-    elif v == "x²":
+    elif key == "x²":
         try:
             st.session_state.expr = str(float(st.session_state.expr) ** 2)
         except:
             st.session_state.expr = "Error"
-    elif v == "log":
+    elif key == "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 = v
-        else:
-            st.session_state.expr += v
-
-    st.rerun()
+        st.session_state.expr += key
 
-# ---------- 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)
+# ---------- DISPLAY ----------
+st.markdown(
+    f"""
+    <div style="
+        background:#111;
+        color:white;
+        padding:18px;
+        font-size:30px;
+        border-radius:12px;
+        text-align:right;
+        margin-bottom:12px;
+    ">
+        {st.session_state.expr or "0"}
+    </div>
+    """,
+    unsafe_allow_html=True
+)
 
-keys = [
+# ---------- BUTTON GRID ----------
+layout = [
     ["C", "√", "x²", "/"],
     ["7", "8", "9", "*"],
     ["4", "5", "6", "-"],
@@ -75,10 +60,9 @@ keys = [
     ["0", ".", "log", "="]
 ]
 
-for row in keys:
+for row in layout:
     cols = st.columns(4)
-    for i, k in enumerate(row):
-        if cols[i].button(k, use_container_width=True):
-            press(k)
-
-st.markdown("</div>", unsafe_allow_html=True)
+    for i, key in enumerate(row):
+        if cols[i].button(key, use_container_width=True):
+            press(key)
+            st.rerun()