Update app.py

app.py

@@ -26,7 +26,8 @@ class DemoRAG:
         self.doc_count = len(self.documents)
         return f"✅ Processed {processed} documents successfully!\n\n📄 Total documents: {self.doc_count}"
     
-    def answer_question(self, question, history):
+    def answer_question(self, message, history):
+        question = message
         if not question.strip():
             return "Please ask a question."
         
@@ -102,10 +103,7 @@ with gr.Blocks(title="Enterprise RAG System", theme=gr.themes.Soft()) as demo:
                         "Can you summarize the key points?",
                         "What does it say about [your topic]?",
                         "Find information about [specific term]"
-                    ],
-                    retry_btn="🔄 Retry",
-                    undo_btn="↶ Undo",
-                    clear_btn="🗑️ Clear"
+                    ]
                 )
         
         # Sidebar