sync with remote

app.py

@@ -3,7 +3,22 @@ import os
 import google.generativeai as genai
 import json
 from PIL import Image
+import re
+import json
 
+# Define constants
+TEXT_PROMPT = """Use the provided document. Read the list of quadratic equations. 
+Return your response as a JSON list. Do not include any extra text, explanations, or backslashes.
+
+Example JSON output:
+[
+  "x^2 - 5x + 6 = 0",
+  "2x^2 + 3x - 1 = 0",
+  "x^2 - 9 = 0",
+  "3x^2 - 2x + 4 = 0",
+  "x^2 + 8x + 15 = 0"
+]
+"""
 
 MODEL_ID = "gemini-2.0-flash-exp"  # Keep the model ID as is
 try:
@@ -17,17 +32,17 @@ except Exception as e:
 model = genai.GenerativeModel(MODEL_ID)
 chat = model.start_chat()
 
-def get_local_pdf_path():
+def get_local_file_path(img_file="problem1.png"):
     """
     Returns the path to the local PDF file.
     """
     try:
-        pdf_path = os.path.join("problems", "problems.pdf")
-        if not os.path.exists(pdf_path):
-            raise FileNotFoundError(f"{pdf_path} does not exist.")
-        return pdf_path
+        file_path = os.path.join("problems", img_file)
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"{file_path} does not exist.")
+        return file_path
     except Exception as e:
-        st.error(f"Failed to find the local PDF: {e}")
+        st.error(f"Failed to find the local file: {e}")
         st.stop()  # Stop if the file is not found
 
 # Initialize conversation history in Streamlit session state
@@ -36,7 +51,7 @@ if "conversation_history" not in st.session_state:
 if "uploaded_file_part" not in st.session_state:  # Store the file *part*
     st.session_state.uploaded_file_part = None
 if "uploaded_pdf_path" not in st.session_state:
-    st.session_state.uploaded_pdf_path = get_local_pdf_path()
+    st.session_state.uploaded_pdf_path = get_local_file_path()
 
 def multimodal_prompt(pdf_path, text_prompt, file_type="PDF"):
     """
@@ -76,7 +91,6 @@ def multimodal_prompt(pdf_path, text_prompt, file_type="PDF"):
     except Exception as e:
         return f"An error occurred: {e}"
 
-
 # --- Main Page ---
 st.title("📚❓Problem Solving Tutor")
 about = """
@@ -84,31 +98,15 @@ about = """
 Replace this placeholder with the actual text.
 """
 
-
-import re
-import json
-
-# Define constants
-TEXT_PROMPT = """Use the provided document. Read the list of 5 quadratic equations. 
-Return your response as a JSON list. Do not include any extra text, explanations, or backslashes.
-
-Example JSON output:
-[
-  "x^2 - 5x + 6 = 0",
-  "2x^2 + 3x - 1 = 0",
-  "x^2 - 9 = 0",
-  "3x^2 - 2x + 4 = 0",
-  "x^2 + 8x + 15 = 0"
-]
-"""
-
-# Main code
-with st.spinner("AI is thinking..."):
+with st.spinner("Loading the problem..."):
     if st.session_state.get("uploaded_pdf_path") is None:
-        st.session_state.uploaded_pdf_path = get_local_pdf_path()
+        st.session_state.uploaded_pdf_path = get_local_file_path("problem1.png")
 
     filepath = st.session_state.uploaded_pdf_path
-    response = multimodal_prompt(filepath, TEXT_PROMPT, file_type="PDF")
+    response = multimodal_prompt(filepath, TEXT_PROMPT, file_type="image")
+
+# --- Display the image ---
+st.image(filepath, caption="Problem Image", use_container_width=True)
 
 # Remove the ```json and ``` and extra spaces.
 try: