Update app.py

app.py

@@ -56,12 +56,12 @@ def add_text(history: List[Tuple[str, str]], text: str) -> List[Tuple[str, str]]
     return history
 
 # Function to get response from the model
-def get_response(history, query, file):
-    if not file:
-        raise gr.Error(message="Upload a PDF")
-    chain = app(file)
+def get_response(history, query):
+    if app.chain is None:
+        raise gr.Error("The chain has not been built yet. Please ensure the vector database is built before querying.")
+    
     try:
-        result = chain.invoke(
+        result = app.chain.invoke(
             {"question": query, "chat_history": app.chat_history}
         )
         app.chat_history.append((query, result["answer"]))
@@ -108,7 +108,7 @@ def set_api_key(api_key):
     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:]}"
+    return f"API Key set to {api_key[:4]}...{api_key[-4:]} and vector database built successfully!"
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -124,15 +124,38 @@ with gr.Blocks() as demo:
                 outputs=[api_key_status]
             )
 
-            with gr.Row():
+            with gr.Tab("Upload PDF"):
                 btn = gr.UploadButton("📁 Upload a PDF", file_types=[".pdf"])
                 show_img = gr.Image(label="Uploaded PDF")
+                
+                btn.upload(
+                    fn=purge_chat_and_render_first,
+                    inputs=[btn],
+                    outputs=[show_img, chatbot],
+                )
+
+            with gr.Tab("Process PDF"):
                 process_btn = gr.Button("Process PDF")
                 show_img_processed = gr.Image(label="Processed PDF")
                 process_status = gr.Textbox(label="Processing Status", interactive=False)
+                
+                process_btn.click(
+                    fn=lambda file: (app.process_file(file), "Processing complete!"),
+                    inputs=[btn],
+                    outputs=[show_img_processed, process_status],
+                )
+
+            with gr.Tab("Build Vector Database"):
                 build_vector_btn = gr.Button("Build Vector Database")
                 status_text = gr.Textbox(label="Status", value="", interactive=False)
-            with gr.Row():
+                
+                build_vector_btn.click(
+                    fn=app.build_chain,
+                    inputs=[btn],
+                    outputs=[status_text],
+                )
+
+            with gr.Tab("Chat"):
                 chatbot = gr.Chatbot(elem_id="chatbot")
                 txt = gr.Textbox(
                     show_label=False,
@@ -143,38 +166,20 @@ with gr.Blocks() as demo:
                 refresh_btn = gr.Button("Refresh Chat", scale=1)
                 source_texts_output = gr.Textbox(label="Source Texts", interactive=False)
 
-        btn.upload(
-            fn=purge_chat_and_render_first,
-            inputs=[btn],
-            outputs=[show_img, chatbot],
-        )
-
-        process_btn.click(
-            fn=lambda file: (app.process_file(file), "Processing complete!"),
-            inputs=[btn],
-            outputs=[show_img_processed, process_status],
-        )
-
-        build_vector_btn.click(
-            fn=app.build_chain,
-            inputs=[btn],
-            outputs=[status_text],
-        )
-
-        submit_btn.click(
-            fn=add_text,
-            inputs=[chatbot, txt],
-            outputs=[chatbot],
-            queue=False,
-        ).success(
-            fn=get_response, inputs=[chatbot, txt, btn], outputs=[chatbot, source_texts_output]
-        )
+                submit_btn.click(
+                    fn=add_text,
+                    inputs=[chatbot, txt],
+                    outputs=[chatbot],
+                    queue=False,
+                ).success(
+                    fn=get_response, inputs=[chatbot, txt], outputs=[chatbot, source_texts_output]
+                )
 
-        refresh_btn.click(
-            fn=refresh_chat,
-            inputs=[],
-            outputs=[chatbot],
-        )
+                refresh_btn.click(
+                    fn=refresh_chat,
+                    inputs=[],
+                    outputs=[chatbot],
+                )
 
     with gr.Tab("Current RAG"):
         with gr.Column():