Update app.py

app.py

@@ -100,15 +100,14 @@ def refresh_chat():
 
 app = MyApp()
 
-# Pre-process the saved PDF file
-saved_file_path = "track_training.pdf"
-with open(saved_file_path, 'rb') as saved_file:
-    app.process_file(saved_file)
-    app.build_chain(saved_file)
-
 # Function to set API key
 def set_api_key(api_key):
     app.set_api_key(api_key)
+    # Pre-process the saved PDF file after setting the API key
+    saved_file_path = "track_training.pdf"
+    with open(saved_file_path, 'rb') as saved_file:
+        app.process_file(saved_file)
+        app.build_chain(saved_file)
     return f"API Key set to {api_key[:4]}...{api_key[-4:]}"
 
 # Gradio interface
@@ -190,7 +189,7 @@ with gr.Blocks() as demo:
             source_texts_output_current = gr.Textbox(label="Source Texts", interactive=False)
 
         def get_response_current(history, query):
-            return get_response(history, query, open(saved_file_path, 'rb'))
+            return get_response(history, query, open("track_training.pdf", 'rb'))
 
         submit_btn_current.click(
             fn=add_text,