Update app.py

app.py

@@ -21,9 +21,7 @@ FAISS_INDEX_PATH = os.path.join(TEMP_DIR, "faiss_index")
 # PDF file path
 PDF_FILE_PATH = "./slide.pdf"
 
-# ---- FIXED: Use the current GA embedding model ----
-# Legacy models (embedding-001, text-embedding-004, text-embedding-005) have
-# all been shut down.  gemini-embedding-001 is the current GA replacement.
+# Current GA embedding model (legacy models have been shut down)
 EMBEDDING_MODEL = "models/gemini-embedding-001"
 
 
@@ -175,6 +173,8 @@ def main():
                 st.session_state["pdf_processed"] = True
                 st.success("PDF processed successfully. You can now ask questions.")
                 st.rerun()
+            except (st.runtime.scriptrunner.RerunException, st.runtime.scriptrunner.StopException):
+                raise
             except FileNotFoundError as e:
                 st.error(str(e))
                 st.stop()
@@ -271,6 +271,8 @@ def main():
                     response = get_response(user_question, api_key, st.session_state["messages"])
                     st.markdown(response)
                     st.session_state["messages"].append({"role": "assistant", "content": response})
+                except (st.runtime.scriptrunner.RerunException, st.runtime.scriptrunner.StopException):
+                    raise
                 except Exception as e:
                     error_msg = f"Error getting answer: {str(e)}"
                     st.error(error_msg)