Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,8 +7,6 @@ from PIL import Image
 import sympy as sp
 import time
 import tempfile
-import pickle
-import pickletools
 
 # Page config
 st.set_page_config(page_title="🎓 Smart Math Teacher", layout="centered")
@@ -63,13 +61,8 @@ try:
     if os.path.exists(zip_path):
         with zipfile.ZipFile(zip_path, "r") as zip_ref:
             zip_ref.extractall(temp_image_folder)
-    else:
-        st.warning(f"Zip file not found: {zip_path}. Using default images if available.")
-except PermissionError as e:
-    st.error(f"Permission error accessing files: {e}")
-    st.info("Using default images without extraction.")
 except Exception as e:
-    st.error(f"Error extracting images: {e}")
+    st.warning(f"Error extracting images: {e}")
 
 if not os.path.exists(dataset_path):
     st.error(f"Dataset not found: {dataset_path}")
@@ -108,31 +101,15 @@ if not subset_df.empty and st.session_state.question_index < len(subset_df):
     if pd.notna(question.get('image')):
         image_name = str(question['image']).strip()
         image_found = False
-        
-        # Check in temp directory first
         for root, _, files in os.walk(temp_image_folder):
             for file in files:
                 if file.lower().startswith(image_name.lower()) or os.path.splitext(file)[0].lower() == image_name.lower():
                     try:
-                        img_path = os.path.join(root, file)
-                        st.image(Image.open(img_path), use_column_width=True)
+                        st.image(Image.open(os.path.join(root, file)), use_column_width=True)
                         image_found = True
                         break
-                    except Exception as e:
-                        st.warning(f"❌ Couldn't load image: {e}")
-        
-        if not image_found:
-            # Fallback to original image folder
-            for root, _, files in os.walk(image_folder):
-                for file in files:
-                    if file.lower().startswith(image_name.lower()) or os.path.splitext(file)[0].lower() == image_name.lower():
-                        try:
-                            st.image(Image.open(os.path.join(root, file)), use_column_width=True)
-                            image_found = True
-                            break
-                        except Exception as e:
-                            st.warning(f"❌ Couldn't load image: {e}")
-        
+                    except:
+                        pass
         if not image_found:
             st.warning("❌ Image not found.")
 
@@ -170,65 +147,11 @@ elif subset_df.empty:
 else:
     st.success("🏁 You've completed all questions in this category!")
 
-# -------------------------------------------------------------
-# ✅ Simple "Ask Any Math Question" Section (no model)
-# -------------------------------------------------------------
-
-qa_data = {
-    "what is the area of a triangle": "Area = (1/2) × base × height",
-    "what is the area of a square": "Area = side²",
-    "what is the area of a circle": "Area = π × radius²",
-    "perimeter of square": "Perimeter = 4 × side",
-    "perimeter of rectangle": "Perimeter = 2 × (length + width)",
-    "volume of cube": "Volume = side³",
-    "volume of sphere": "Volume = (4/3) × π × radius³",
-    "what is pi": "π ≈ 3.14159",
-    "how to calculate speed": "Speed = Distance ÷ Time",
-    "what is the derivative of x squared": "The derivative of x² is 2x."
-}
-
-# Store QA data using pickle (optional demo)
-try:
-    with open('qa_data.pkl', 'wb') as f:
-        pickle.dump(qa_data, f)
-    with open('qa_data.pkl', 'rb') as f:
-        pickletools.dis(f)
-except Exception as e:
-    st.warning(f"Could not create pickle file: {e}")
-
-st.markdown("## 💡 Ask Any Math Question")
-question_input = st.text_input("Ask a question like 'What is the area of a circle?' or 'Derivative of x^2'")
-
-if question_input:
-    # simple lookup instead of embeddings
-    found = None
-    for q in qa_data:
-        if q in question_input.lower():
-            found = qa_data[q]
-            break
-
-    if found:
-        st.success(f"🤖 Answer:\n\n**{found}**")
-    else:
-        try:
-            if "derivative" in question_input.lower():
-                expr = question_input.lower().split("of")[-1].strip()
-                var = sp.symbols('x')
-                derivative = sp.diff(sp.sympify(expr.replace("^", "**")), var)
-                st.success(f"🧮 Derivative of {expr} is:\n\n**{derivative}**")
-            else:
-                result = sp.sympify(question_input.replace("^", "**"))
-                simplified = sp.simplify(result)
-                st.success(f"✅ Answer: {simplified}")
-        except Exception as e:
-            st.warning(f"⚠️ Couldn't understand the question. Error: {e}")
-
-# Cleanup function (optional)
+# Cleanup function
 def cleanup():
     import shutil
     if 'temp_dir' in st.session_state and os.path.exists(st.session_state.temp_dir):
         shutil.rmtree(st.session_state.temp_dir)
 
-# Register cleanup when app closes
 import atexit
 atexit.register(cleanup)