Update app.py

app.py

@@ -1,30 +1,134 @@
+# import streamlit as st
+# import sympy as sp
+# import pandas as pd
+
+# st.set_page_config(page_title="Logic Solver", layout="centered")
+
+# st.title("🧩 AI Logic Solver")
+# st.write("Enter a logical expression using:")
+# st.write("~ (NOT), & (AND), | (OR), >> (IMPLIES)")
+
+# # User input
+# expr_input = st.text_input("Enter expression (e.g., ~(p & q) >> r):")
+
+# if expr_input:
+#     try:
+#         # Define symbols
+#         p, q, r = sp.symbols('p q r')
+
+#         # Convert string to expression
+#         expr = sp.sympify(expr_input)
+
+#         st.subheader("📌 Simplified Expression:")
+#         simplified = sp.simplify_logic(expr)
+#         st.write(simplified)
+
+#         # Truth table
+#         st.subheader("📊 Truth Table:")
+
+#         variables = sorted(expr.free_symbols, key=lambda x: str(x))
+#         rows = []
+
+#         for values in range(2**len(variables)):
+#             combination = list(map(int, bin(values)[2:].zfill(len(variables))))
+#             subs = dict(zip(variables, combination))
+#             result = expr.subs(subs)
+
+#             row = {str(var): val for var, val in subs.items()}
+#             row["Result"] = int(bool(result))
+#             rows.append(row)
+
+#         df = pd.DataFrame(rows)
+#         st.dataframe(df)
+
+#     except Exception as e:
+#         st.error("❌ Invalid expression. Please follow correct syntax.")
 import streamlit as st
 import sympy as sp
 import pandas as pd
 
-st.set_page_config(page_title="Logic Solver", layout="centered")
+# Page config
+st.set_page_config(page_title="AI Logic Solver", layout="centered")
 
+# Title
 st.title("🧩 AI Logic Solver")
-st.write("Enter a logical expression using:")
-st.write("~ (NOT), & (AND), | (OR), >> (IMPLIES)")
+st.markdown("Solve logical expressions with **steps + truth table**")
+
+# Instructions
+st.info("Use: ~ (NOT), & (AND), | (OR), >> (IMPLIES)")
+
+# Quick input buttons
+col1, col2, col3, col4 = st.columns(4)
 
-# User input
-expr_input = st.text_input("Enter expression (e.g., ~(p & q) >> r):")
+if "expr" not in st.session_state:
+    st.session_state.expr = ""
 
+if col1.button("p"):
+    st.session_state.expr += "p"
+if col2.button("q"):
+    st.session_state.expr += "q"
+if col3.button("r"):
+    st.session_state.expr += "r"
+if col4.button("Clear"):
+    st.session_state.expr = ""
+
+col5, col6, col7, col8 = st.columns(4)
+
+if col5.button("~"):
+    st.session_state.expr += "~"
+if col6.button("&"):
+    st.session_state.expr += " & "
+if col7.button("|"):
+    st.session_state.expr += " | "
+if col8.button(">>"):
+    st.session_state.expr += " >> "
+
+# Input box
+expr_input = st.text_input("Enter expression:", value=st.session_state.expr)
+
+# Process
 if expr_input:
     try:
         # Define symbols
         p, q, r = sp.symbols('p q r')
 
-        # Convert string to expression
+        # Convert expression
         expr = sp.sympify(expr_input)
 
-        st.subheader("📌 Simplified Expression:")
-        simplified = sp.simplify_logic(expr)
+        st.success("✅ Expression parsed successfully!")
+
+        # =========================
+        # STEP-BY-STEP (Basic)
+        # =========================
+        st.subheader("🧠 Step-by-Step Simplification")
+
+        steps = []
+
+        # Step 1: Original
+        steps.append(f"Original: {expr}")
+
+        # Step 2: Remove implication
+        expr_no_impl = expr.replace(
+            lambda x: isinstance(x, sp.Implies),
+            lambda x: ~x.args[0] | x.args[1]
+        )
+        steps.append(f"Remove implication: {expr_no_impl}")
+
+        # Step 3: Simplify
+        simplified = sp.simplify_logic(expr_no_impl)
+        steps.append(f"Simplified: {simplified}")
+
+        for step in steps:
+            st.write("➡️", step)
+
+        # Final simplified
+        st.subheader("📌 Final Simplified Expression:")
         st.write(simplified)
 
-        # Truth table
-        st.subheader("📊 Truth Table:")
+        # =========================
+        # TRUTH TABLE
+        # =========================
+        st.subheader("📊 Truth Table")
 
         variables = sorted(expr.free_symbols, key=lambda x: str(x))
         rows = []
@@ -32,14 +136,24 @@ if expr_input:
         for values in range(2**len(variables)):
             combination = list(map(int, bin(values)[2:].zfill(len(variables))))
             subs = dict(zip(variables, combination))
-            result = expr.subs(subs)
+
+            result = bool(expr.subs(subs))  # ✅ FIXED
 
             row = {str(var): val for var, val in subs.items()}
-            row["Result"] = int(bool(result))
+            row["Result"] = int(result)
             rows.append(row)
 
         df = pd.DataFrame(rows)
-        st.dataframe(df)
+        st.dataframe(df, use_container_width=True)
+
+        # Download option
+        st.download_button(
+            "📥 Download Truth Table",
+            df.to_csv(index=False),
+            file_name="truth_table.csv",
+            mime="text/csv"
+        )
 
     except Exception as e:
-        st.error("❌ Invalid expression. Please follow correct syntax.")
+        st.error(f"❌ Invalid expression: {str(e)}")
+        st.info("💡 Example: ~(p & q) >> r")