Update app.py

app.py

@@ -65,18 +65,22 @@ def get_response(history, query, file):
     if not file:
         raise gr.Error(message="Upload a PDF")
     chain = app(file)
-    result = chain(
-        {"question": query, "chat_history": app.chat_history}, return_only_outputs=True
-    )
-    app.chat_history.append((query, result["answer"]))
-    source_docs = result["source_documents"]
-    source_texts = []
-    for doc in source_docs:
-        source_texts.append(f"Page {doc.metadata['page'] + 1}: {doc.page_content}")
-    source_texts_str = "\n\n".join(source_texts)
-    for char in result["answer"]:
-        history[-1][-1] += char
-        yield history, "", source_texts_str
+    try:
+        result = chain.invoke(
+            {"question": query, "chat_history": app.chat_history}
+        )
+        app.chat_history.append((query, result["answer"]))
+        source_docs = result["source_documents"]
+        source_texts = []
+        for doc in source_docs:
+            source_texts.append(f"Page {doc.metadata['page'] + 1}: {doc.page_content}")
+        source_texts_str = "\n\n".join(source_texts)
+        for char in result["answer"]:
+            history[-1][-1] += char
+            yield history, "", source_texts_str
+    except Exception:
+        app.chat_history.append((query, "I have no information about it. Feed me knowledge, please!"))
+        yield history, "", "I have no information about it. Feed me knowledge, please!"
 
 def render_file(file) -> Image.Image:
     doc = fitz.open(file.name)
@@ -107,6 +111,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Step 2: Process File"):
         process_btn = gr.Button("Process PDF")
         show_img_processed = gr.Image(label="Processed PDF")
+        process_status = gr.Textbox(label="Processing Status", interactive=False)
 
     with gr.Tab("Step 3: Build Vector Database"):
         build_vector_btn = gr.Button("Build Vector Database")
@@ -130,9 +135,9 @@ with gr.Blocks() as demo:
     )
 
     process_btn.click(
-        fn=app.process_file,
+        fn=lambda file: (app.process_file(file), "Processing complete!"),
         inputs=[btn],
-        outputs=[show_img_processed],
+        outputs=[show_img_processed, process_status],
     )
 
     build_vector_btn.click(