Update app.py

app.py

@@ -203,15 +203,18 @@ def create_gradio_interface(rag_system: AdvancedRAGSystem) -> gr.Blocks:
         except Exception as e:
             return f"Error: {str(e)}"
 
-    def query_fin(question):
-        """Query system and update history with error handling"""
+    def query_streaming(question: str) -> Generator[str, None, None]:
         try:
-            yield rag_system.query(question).choices[0].delta.content
+            for response in rag_system.query(question):
+                yield response
         except Exception as e:
-            pass
+            yield f"Error: {str(e)}"
             
     def update_history(question: str):
-        return f"Last context used ({rag_system.source_documents} documents):\n\n{rag_system.context}"
+        try:
+            return f"Last context used ({rag_system.source_documents} documents):\n\n{rag_system.context}"
+        except Exception as e:
+            return f"Error retrieving context: {str(e)}"
         
     with gr.Blocks(title="Advanced RAG System") as demo:
         gr.Markdown("# Advanced RAG System with PDF Processing")