Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import shutil
 import logging
 from typing import List, Dict
 import torch
@@ -27,17 +28,19 @@ class DocumentManager:
     
     def __init__(self):
         self.upload_folder = UPLOAD_FOLDER
+        if os.path.exists(self.upload_folder):
+            shutil.rmtree(self.upload_folder)
         os.makedirs(self.upload_folder, exist_ok=True)
         self.max_files = 5
         self.max_file_size = 10 * 1024 * 1024  # 10 MB
         self.supported_formats = ['.pdf', '.txt', '.docx']
         self.documents = []
         
-    def validate_file(self, file):
-        if os.path.getsize(file.name) > self.max_file_size:
+    def validate_file(self, file_path, file_size):
+        if file_size > self.max_file_size:
             raise ValueError(f"File size exceeds {self.max_file_size // 1024 // 1024}MB limit")
         
-        ext = os.path.splitext(file.name)[1].lower()
+        ext = os.path.splitext(file_path)[1].lower()
         if ext not in self.supported_formats:
             raise ValueError(f"Unsupported file format. Supported formats: {', '.join(self.supported_formats)}")
             
@@ -65,22 +68,34 @@ class DocumentManager:
             logger.error(f"Error loading {file_path}: {str(e)}")
             raise
 
-    def process_upload(self, files: List) -> str:
-        if len(os.listdir(self.upload_folder)) + len(files) > self.max_files:
-            raise ValueError(f"Maximum number of documents ({self.max_files}) exceeded")
+    def process_upload(self, files: List[gr.File]) -> str:
+        if not files:
+            return "No files uploaded"
+            
+        current_files = len(os.listdir(self.upload_folder))
+        if current_files + len(files) > self.max_files:
+            return f"Maximum number of documents ({self.max_files}) exceeded"
             
         processed_files = []
         for file in files:
             try:
-                self.validate_file(file)
-                save_path = os.path.join(self.upload_folder, file.name)
-                file.save(save_path)
+                file_path = file.name
+                file_size = os.path.getsize(file_path)
+                
+                self.validate_file(file_path, file_size)
+                
+                # Copy file to upload folder
+                filename = os.path.basename(file_path)
+                save_path = os.path.join(self.upload_folder, filename)
+                shutil.copy2(file_path, save_path)
+                
                 docs = self.load_document(save_path)
                 self.documents.extend(docs)
-                processed_files.append(file.name)
+                processed_files.append(filename)
+                
             except Exception as e:
-                logger.error(f"Error processing {file.name}: {str(e)}")
-                return f"Error processing {file.name}: {str(e)}"
+                logger.error(f"Error processing {file_path}: {str(e)}")
+                return f"Error processing {os.path.basename(file_path)}: {str(e)}"
                 
         return f"Successfully processed files: {', '.join(processed_files)}"
 
@@ -203,7 +218,7 @@ def process_file_upload(files):
     """Handle file uploads and system initialization."""
     try:
         upload_result = rag_system.document_manager.process_upload(files)
-        if "Error" in upload_result:
+        if "Error" in upload_result or "Maximum" in upload_result:
             return upload_result
             
         init_result = rag_system.initialize_system(rag_system.document_manager.documents)
@@ -231,7 +246,16 @@ def process_query(message, history):
         return history + [(message, f"Error: {str(e)}")]
 
 # Create Gradio interface
-demo = gr.Blocks(css="div.gradio-container {background-color: #f0f2f6}")
+demo = gr.Blocks(css="""
+    div.gradio-container {background-color: #f0f2f6}
+    div.upload-box { 
+        border: 2px dashed #ccc;
+        padding: 20px;
+        border-radius: 10px;
+        background-color: #ffffff;
+        margin-bottom: 15px;
+    }
+""")
 
 with demo:
     gr.HTML("""
@@ -244,29 +268,42 @@ with demo:
     """)
     
     with gr.Row():
-        file_output = gr.File(
-            file_count="multiple",
-            label="Upload Documents (PDF, TXT, DOCX - Max 5 files, 10MB each)"
-        )
-    
-    upload_button = gr.Button("Upload and Initialize")
-    system_output = gr.Textbox(label="System Status")
-    
-    chatbot = gr.Chatbot(
-        show_label=False,
-        container=True,
-        height=400,
-        show_copy_button=True
-    )
-    
-    with gr.Row():
-        message = gr.Textbox(
-            placeholder="Ask a question about your documents...",
-            show_label=False,
-            container=False,
-            scale=8
-        )
-        clear = gr.Button("🗑️ Clear", size="sm", scale=1)
+        # Sidebar for document upload
+        with gr.Column(scale=1):
+            with gr.Box(elem_classes="upload-box"):
+                gr.HTML("<h3>📁 Document Upload</h3>")
+                file_output = gr.File(
+                    file_count="multiple",
+                    label="Upload Documents (PDF, TXT, DOCX)",
+                    elem_classes="file-upload"
+                )
+                gr.HTML("""
+                    <div style="font-size: 0.8em; color: #666;">
+                        <p>• Maximum 5 files</p>
+                        <p>• 10MB per file</p>
+                        <p>• Supported formats: PDF, TXT, DOCX</p>
+                    </div>
+                """)
+                upload_button = gr.Button("📤 Upload and Initialize", variant="primary")
+                system_output = gr.Textbox(label="System Status", interactive=False)
+        
+        # Main chat area
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(
+                show_label=False,
+                container=True,
+                height=600,
+                show_copy_button=True
+            )
+            
+            with gr.Row():
+                message = gr.Textbox(
+                    placeholder="Ask a question about your documents...",
+                    show_label=False,
+                    container=False,
+                    scale=8
+                )
+                clear = gr.Button("🗑️ Clear", size="sm", scale=1)
     
     gr.HTML("""
         <div style="text-align: center; max-width: 800px; margin: 20px auto; padding: 20px;