Update app.py

app.py

@@ -96,7 +96,7 @@ class DocumentAIBot:
         # Create vector store
         vector_store = FAISS.from_documents(chunks, self.embedding_model)
         return vector_store
-    
+
     def setup_retrieval_chain(self, vector_store):
         """Set up the retrieval chain with the vector store."""
         retriever = vector_store.as_retriever(
@@ -161,35 +161,19 @@ def generate_session_id():
     import uuid
     return str(uuid.uuid4())
 
-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, os.path.basename(file.name))
-    
-    with open(temp_path, "wb") as f:
-        f.write(file.read())
-    
-    return temp_path
-
-def clear_conversation():
-    """Clear the conversation history for the current session."""
-    global conversation_history, current_session_id
-    
-    if current_session_id and current_session_id in conversation_history:
-        conversation_history[current_session_id] = []
-    
-    return [], f"Conversation cleared. You can continue asking questions about '{current_document_name}'."
-
-def process_uploaded_document(file):
+def process_uploaded_document(file_path):
     """Process an uploaded document and set up the session."""
     global current_session_id, current_document_store, current_document_name, conversation_history
     
     try:
-        if file is None:
+        if file_path is None:
             return None, "Please upload a document first."
         
-        # Save the uploaded file
-        file_path = save_uploaded_file(file)
+        # In newer Gradio versions, the file input with type="filepath" returns the path directly
+        # No need to save the file as it's already saved by Gradio
+        
+        # Extract filename for display
+        filename = os.path.basename(file_path)
         
         # Create document AI bot if not already created
         if not hasattr(process_uploaded_document, "bot"):
@@ -205,24 +189,33 @@ def process_uploaded_document(file):
         # Update global variables
         current_session_id = session_id
         current_document_store = vector_store
-        current_document_name = os.path.basename(file.name)
+        current_document_name = filename
         
-        return [], f"Document '{os.path.basename(file.name)}' processed successfully. You can now ask questions about it."
+        return [], f"Document '{filename}' processed successfully. You can now ask questions about it."
     
     except Exception as e:
         import traceback
         traceback.print_exc()
         return None, f"Error processing document: {str(e)}"
 
+def clear_conversation():
+    """Clear the conversation history for the current session."""
+    global conversation_history, current_session_id
+    
+    if current_session_id and current_session_id in conversation_history:
+        conversation_history[current_session_id] = []
+    
+    return [], f"Conversation cleared. You can continue asking questions about '{current_document_name}'."
+
 def answer_question(question, history):
     """Answer a question about the current document."""
     global current_session_id, current_document_store, conversation_history
     
     if not current_document_store:
-        return "Please upload a document first.", history
+        return "", history + [(question, "Please upload a document first.")]
     
     if not hasattr(process_uploaded_document, "bot"):
-        return "Document AI bot not initialized. Please reload the page and try again.", history
+        return "", history + [(question, "Document AI bot not initialized. Please reload the page and try again.")]
     
     try:
         # Get current chat history
@@ -261,7 +254,7 @@ def build_interface():
                 file_input = gr.File(
                     label="Upload Document",
                     file_types=[".pdf", ".txt", ".docx", ".pptx"],
-                    type="filepath"  # Changed from "file" to "filepath"
+                    type="filepath"  # This returns the file path directly
                 )
                 
                 upload_button = gr.Button("Process Document", variant="primary")