Update app.py

app.py

@@ -164,7 +164,7 @@ def generate_session_id():
 def save_uploaded_file(file):
     """Save uploaded file to a temporary location and return the path."""
     temp_dir = tempfile.gettempdir()
-    temp_path = os.path.join(temp_dir, file.name)
+    temp_path = os.path.join(temp_dir, os.path.basename(file.name))
     
     with open(temp_path, "wb") as f:
         f.write(file.read())
@@ -205,9 +205,9 @@ def process_uploaded_document(file):
         # Update global variables
         current_session_id = session_id
         current_document_store = vector_store
-        current_document_name = file.name
+        current_document_name = os.path.basename(file.name)
         
-        return [], f"Document '{file.name}' processed successfully. You can now ask questions about it."
+        return [], f"Document '{os.path.basename(file.name)}' processed successfully. You can now ask questions about it."
     
     except Exception as e:
         import traceback
@@ -219,10 +219,10 @@ def answer_question(question, history):
     global current_session_id, current_document_store, conversation_history
     
     if not current_document_store:
-        return "Please upload a document first."
+        return "Please upload a document first.", history
     
     if not hasattr(process_uploaded_document, "bot"):
-        return "Document AI bot not initialized. Please reload the page and try again."
+        return "Document AI bot not initialized. Please reload the page and try again.", history
     
     try:
         # Get current chat history
@@ -239,12 +239,14 @@ def answer_question(question, history):
         # Update conversation history
         conversation_history[current_session_id] = updated_history
         
-        return answer
+        # Update the display history
+        history = history + [(question, answer)]
+        return "", history
     
     except Exception as e:
         import traceback
         traceback.print_exc()
-        return f"Error generating answer: {str(e)}"
+        return "", history + [(question, f"Error generating answer: {str(e)}")]
 
 def build_interface():
     """Build and launch the Gradio interface."""
@@ -259,7 +261,7 @@ def build_interface():
                 file_input = gr.File(
                     label="Upload Document",
                     file_types=[".pdf", ".txt", ".docx", ".pptx"],
-                    type="file"
+                    type="filepath"  # Changed from "file" to "filepath"
                 )
                 
                 upload_button = gr.Button("Process Document", variant="primary")
@@ -289,8 +291,6 @@ def build_interface():
                         label="Ask a question about the document",
                         placeholder="What is the main topic of this document?",
                         lines=2,
-                        max_lines=5,
-                        interactive=True,
                         show_label=True
                     )
                     
@@ -306,21 +306,13 @@ def build_interface():
         submit_button.click(
             answer_question,
             inputs=[question_input, chatbot],
-            outputs=chatbot
-        ).then(
-            lambda: "",
-            None,
-            question_input
+            outputs=[question_input, chatbot]
         )
         
         question_input.submit(
             answer_question,
             inputs=[question_input, chatbot],
-            outputs=chatbot
-        ).then(
-            lambda: "",
-            None,
-            question_input
+            outputs=[question_input, chatbot]
         )
         
         clear_button.click(
@@ -328,16 +320,6 @@ def build_interface():
             inputs=[],
             outputs=[chatbot, upload_status]
         )
-        
-        # Add CSS for better styling
-        interface.load(
-            js="""
-            () => {
-                document.querySelector('body').style.backgroundColor = '#f7f7f7';
-                document.querySelector('.gradio-container').style.maxWidth = '1200px';
-            }
-            """
-        )
     
     return interface
 
@@ -345,9 +327,7 @@ def build_interface():
 if __name__ == "__main__":
     demo = build_interface()
     demo.launch(
-        share=True,
         server_name="0.0.0.0",
         server_port=7860,
-        debug=True,
-        show_api=False
+        share=False
     )