Upload app.py

app.py

@@ -6,6 +6,7 @@ from typing import List, Tuple
 import tempfile
 import shutil
 from dotenv import load_dotenv
+import time
 
 # Load environment variables
 load_dotenv()
@@ -25,8 +26,8 @@ pc = Pinecone(api_key=pinecone_api_key)
 UPLOAD_FOLDER = "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
-def process_files(files, *metadata_inputs):
-    """Process multiple files with individual metadata for each file"""
+def process_files_with_progress(files, *metadata_inputs, progress=gr.Progress()):
+    """Process multiple files with individual metadata and show progress"""
     if not files:
         return "❌ Error: No files selected", ""
     
@@ -36,8 +37,11 @@ def process_files(files, *metadata_inputs):
     try:
         results = []
         errors = []
+        total_files = len(files)
         
         # Initialize Pinecone Assistant
+        progress(0, desc="🔧 Initializing Pinecone Assistant...")
+        time.sleep(0.5)  # Small delay to show the progress
         assistant_name = os.getenv("PINECONE_ASSISTANT_NAME", "gstminutes")
         assistant = pc.assistant.Assistant(assistant_name=assistant_name)
 
@@ -45,6 +49,7 @@ def process_files(files, *metadata_inputs):
         for i, file_path in enumerate(files):
             try:
                 filename = os.path.basename(file_path)
+                progress((i / total_files), desc=f"📄 Processing {filename}... ({i+1}/{total_files})")
                 
                 # Get metadata for this specific file (3 fields per file: sections, keywords, description)
                 sections_idx = i * 3
@@ -67,6 +72,7 @@ def process_files(files, *metadata_inputs):
                     continue
                 
                 # Prepare metadata for this file
+                progress((i / total_files), desc=f"🏷️ Preparing metadata for {filename}...")
                 metadata = {
                     "sections": [s.strip() for s in sections.split(",") if s.strip()],
                     "keywords": [k.strip() for k in keywords.split(",") if k.strip()],
@@ -74,10 +80,12 @@ def process_files(files, *metadata_inputs):
                 }
                 
                 # Copy to uploads directory
+                progress((i / total_files), desc=f"📁 Copying {filename} to uploads...")
                 destination_path = os.path.join(UPLOAD_FOLDER, filename)
                 shutil.copy2(file_path, destination_path)
 
                 # Upload to Pinecone Assistant
+                progress((i / total_files), desc=f"☁️ Uploading {filename} to Pinecone...")
                 response = assistant.upload_file(
                     file_path=destination_path, 
                     metadata=metadata, 
@@ -97,6 +105,10 @@ def process_files(files, *metadata_inputs):
                     "error": f"❌ Error: {str(file_error)}"
                 })
 
+        # Final progress update
+        progress(1.0, desc="✅ Processing complete!")
+        time.sleep(0.5)
+
         # Format results for display
         success_count = len(results)
         error_count = len(errors)
@@ -122,11 +134,11 @@ def process_files(files, *metadata_inputs):
             for error in errors:
                 detailed_results += f"- **{error['filename']}** - {error['error']}\n"
         
-        return status_message, detailed_results
+        return status_message, detailed_results, "✅ **Processing completed successfully!**"
     
     except Exception as e:
         error_msg = f"❌ **Critical Error:** {str(e)}"
-        return error_msg, ""
+        return error_msg, "", "❌ **Processing failed with error**"
 
 def update_metadata_fields(files):
     """Update metadata fields based on uploaded files"""
@@ -156,7 +168,15 @@ def update_metadata_fields(files):
 
 def clear_form():
     """Clear all form fields"""
-    return [None] + [""] * 30 + ["", ""]
+    return [None] + [""] * 30 + ["", "", "🟢 **Ready to process documents**"]
+
+def start_processing():
+    """Show processing started status"""
+    return "🔄 **Processing documents... Please wait**"
+
+def finish_processing():
+    """Show processing finished status"""
+    return "✅ **Processing completed successfully!**"
 
 # Create Gradio interface
 with gr.Blocks(
@@ -235,6 +255,13 @@ with gr.Blocks(
                 size="lg"
             )
     
+    # Processing status indicator
+    with gr.Row():
+        processing_status = gr.Markdown(
+            value="🟢 **Ready to process documents**",
+            visible=True
+        )
+    
     gr.Markdown("---")
     
     # Results section
@@ -261,15 +288,19 @@ with gr.Blocks(
         outputs=metadata_fields
     )
     
+    # Show processing status when upload starts
     upload_btn.click(
-        fn=process_files,
+        fn=start_processing,
+        outputs=[processing_status]
+    ).then(
+        fn=process_files_with_progress,
         inputs=[files_input] + metadata_fields,
-        outputs=[status_output, results_output]
+        outputs=[status_output, results_output, processing_status]
     )
     
     clear_btn.click(
         fn=clear_form,
-        outputs=[files_input] + metadata_fields + [status_output, results_output]
+        outputs=[files_input] + metadata_fields + [status_output, results_output, processing_status]
     )
     
     # Footer