Update app.py

app.py

@@ -76,21 +76,22 @@ Answer:"""
         except Exception as e:
             return f"Error processing PDF: {str(e)}"
 
-    def generate_response(self, query):
+    def generate_response(self, message, history):
         try:
-            if not query.strip():
-                return "Please enter a question."
-                
             if self.vector_store is None:
                 return "Please upload and process a PDF first."
 
+            query = message.strip()
+            if not query:
+                return "Please enter a question."
+
             # Search for relevant chunks
             relevant_chunks = self.vector_store.similarity_search(query, k=3)
             context = "\n\n".join([doc.page_content for doc in relevant_chunks])
 
             # Format conversation history
             conversation_history = "\n".join([
-                f"Q: {q}\nA: {a}" for q, a in self.conversation_history[-3:]  # Keep last 3 exchanges
+                f"Q: {q}\nA: {a}" for q, a in history[-3:] if q and a  # Keep last 3 exchanges
             ])
 
             # Create prompt with system prompt, context, and conversation history
@@ -113,9 +114,6 @@ Answer:"""
             ):
                 response += message.choices[0].delta.content
 
-            # Update conversation history
-            self.conversation_history.append((query, response))
-            
             return response
         except Exception as e:
             return f"Error generating response: {str(e)}"
@@ -129,40 +127,45 @@ def create_gradio_interface():
         gr.Markdown("# PDF Question Answering System")
         
         with gr.Row():
-            with gr.Column():
-                pdf_input = gr.File(
-                    label="Upload PDF",
-                    file_types=[".pdf"],
-                    type="filepath"
-                )
-                process_button = gr.Button("Process PDF")
-                status_output = gr.Textbox(label="Status", interactive=False)
-            
-        with gr.Row():
-            with gr.Column():
-                query_input = gr.Textbox(
-                    label="Ask a question about the PDF",
-                    placeholder="Enter your question here..."
-                )
-                submit_button = gr.Button("Submit")
-                answer_output = gr.Textbox(
-                    label="Answer",
-                    interactive=False,
-                    lines=5
-                )
-
+            pdf_input = gr.File(
+                label="Upload PDF",
+                file_types=[".pdf"],
+                type="filepath"
+            )
+            process_button = gr.Button("Process PDF")
+            status_output = gr.Textbox(label="Status", interactive=False)
+        
         process_button.click(
             fn=rag.process_pdf,
             inputs=[pdf_input],
             outputs=[status_output]
         )
-        
-        submit_button.click(
+
+        chatbot = gr.ChatInterface(
             fn=rag.generate_response,
-            inputs=[query_input],
-            outputs=[answer_output]
+            chatbot=gr.Chatbot(
+                label="Chat History",
+                height=400,
+                show_label=True,
+            ),
+            textbox=gr.Textbox(
+                placeholder="Ask a question about the PDF...",
+                container=False,
+                scale=7,
+            ),
+            title="Chat with your PDF",
+            description="Upload a PDF and ask questions about its contents.",
+            theme="soft",
+            examples=[
+                "What is the main topic of this document?",
+                "Can you summarize the key points?",
+                "What are the main conclusions?",
+            ],
+            retry_btn="Retry",
+            undo_btn="Undo",
+            clear_btn="Clear",
         )
-    
+
     return demo
 
 if __name__ == "__main__":