Update app.py

app.py

@@ -399,7 +399,7 @@ with gr.Blocks(title="Document Q&A System", css=custom_css) as demo:
                     label="Upload Document",
                     file_types=[".pdf", ".txt", ".doc", ".docx"],
                     file_count="single",
-                    type="binary"  # Specify that we want binary data
+                    type="binary"
                 )
                 file_output = gr.Textbox(
                     label="Upload Status",
@@ -410,28 +410,46 @@ with gr.Blocks(title="Document Q&A System", css=custom_css) as demo:
 
     with gr.Row():
         with gr.Column(scale=2):
-            with gr.Group(elem_classes="chat-container"):
-                chatbot = gr.Chatbot(
-                    height=500,
-                    type="messages",
-                    show_label=False,
-                    elem_classes="chatbot"
-                )
-                with gr.Row():
-                    question = gr.Textbox(
-                        label="Ask a question",
-                        placeholder="Type your question here...",
-                        show_label=False,
-                        container=False,
-                        elem_classes="textbox"
-                    )
-                    submit_btn = gr.Button(
-                        "Send",
-                        variant="primary",
-                        elem_classes="send-button"
-                    )
-                question.submit(process_question, [question, chatbot], [question, chatbot])
-                submit_btn.click(process_question, [question, chatbot], [question, chatbot])
+            chatbot = gr.Chatbot(
+                height=500,
+                type="messages",
+                show_label=False,
+                elem_classes="chatbot",
+                placeholder="Ask questions about your uploaded document..."
+            )
+            
+            def chat_fn(message, history):
+                if not rag.thread_id:
+                    return "Please upload a document first."
+                try:
+                    response = rag.ask_question(message)
+                    return response
+                except Exception as e:
+                    return f"Error: {str(e)}"
+
+            def vote(data: gr.LikeData):
+                if data.liked:
+                    print("You upvoted this response: " + data.value["value"])
+                else:
+                    print("You downvoted this response: " + data.value["value"])
+
+            chatbot.like(vote, None, None)
+            
+            gr.ChatInterface(
+                fn=chat_fn,
+                chatbot=chatbot,
+                examples=[
+                    "What is this document about?",
+                    "Can you summarize the main points?",
+                    "What are the key findings?"
+                ],
+                title="Document Q&A",
+                description="Ask questions about your uploaded document",
+                theme="soft",
+                retry_btn=None,
+                undo_btn=None,
+                clear_btn="Clear Chat"
+            )
 
     with gr.Row():
         with gr.Column(scale=1):