Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 from pathlib import Path
-from pinecone import Pinecone, ApiException
+from pinecone import Pinecone
 from typing import List, Tuple
 import tempfile
 import shutil
@@ -24,41 +24,29 @@ if missing_vars:
 pinecone_api_key = os.getenv("PINECONE_API_KEY")
 pc = Pinecone(api_key=pinecone_api_key)
 
-# Get the Pinecone Assistant name from environment variables
-PINE_ASSISTANT_NAME = os.getenv("PINECONE_ASSISTANT_NAME", "gstminutes")
-
-# Initialize Pinecone Assistant (make this a global object for re-use)
-try:
-    pinecone_assistant = pc.assistant.Assistant(assistant_name=PINE_ASSISTANT_NAME)
-except Exception as e:
-    print(f"Error initializing Pinecone Assistant: {e}")
-    pinecone_assistant = None
-
 # Create uploads directory
 UPLOAD_FOLDER = "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 def list_uploaded_files(progress=gr.Progress()):
     """List all files uploaded to Pinecone Assistant with their metadata and timestamps"""
-    if not pinecone_assistant:
-        error_msg = f"❌ **Error:** Pinecone Assistant not initialized. Check your API key and assistant name."
-        return error_msg, ""
-        
     try:
         progress(0.1, desc="🔍 Connecting to Pinecone Assistant...")
+        assistant_name = os.getenv("PINECONE_ASSISTANT_NAME", "gstminutes")
+        assistant = pc.assistant.Assistant(assistant_name=assistant_name)
         
         progress(0.3, desc="📋 Fetching file list...")
         time.sleep(0.5)
         
         # List all files in the assistant
-        files_response = pinecone_assistant.list_files()
+        files_response = assistant.list_files()
         
         progress(0.7, desc="📊 Processing file information...")
         time.sleep(0.3)
         
         if not files_response or not hasattr(files_response, 'files') or not files_response.files:
             progress(1.0, desc="✅ Complete - No files found")
-            return f"📋 **No files found in Pinecone Assistant '{PINE_ASSISTANT_NAME}'**", ""
+            return "📋 **No files found in Pinecone Assistant**", ""
         
         files_list = files_response.files
         total_files = len(files_list)
@@ -69,7 +57,7 @@ def list_uploaded_files(progress=gr.Progress()):
         progress(0.9, desc="📝 Formatting results...")
         
         # Create summary
-        summary = f"📊 **Files Summary for Assistant '{PINE_ASSISTANT_NAME}'**\n\n"
+        summary = f"📊 **Files Summary**\n\n"
         summary += f"📁 **Total files:** {total_files}\n"
         summary += f"🕐 **Last updated:** {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
         
@@ -117,7 +105,7 @@ def list_uploaded_files(progress=gr.Progress()):
                 # Try to get metadata if available
                 try:
                     # Get file details for metadata
-                    file_details = pinecone_assistant.describe_file(file_id=file_id)
+                    file_details = assistant.describe_file(file_id=file_id)
                     if hasattr(file_details, 'metadata') and file_details.metadata:
                         metadata = file_details.metadata
                         detailed_info += f"- **🏷️ Metadata:**\n"
@@ -144,9 +132,6 @@ def list_uploaded_files(progress=gr.Progress()):
         
         return summary, detailed_info
         
-    except ApiException as e:
-        error_msg = f"❌ **Pinecone API Error:** {e.body.decode('utf-8')}"
-        return error_msg, ""
     except Exception as e:
         error_msg = f"❌ **Error retrieving file list:** {str(e)}"
         return error_msg, ""
@@ -157,20 +142,23 @@ def refresh_file_list():
 
 def process_files_with_progress(files, *metadata_inputs, progress=gr.Progress()):
     """Process multiple files with individual metadata and show progress"""
-    if not pinecone_assistant:
-        return f"❌ Error: Pinecone Assistant not initialized. Check your environment variables.", "", "❌ **Processing failed**"
-    
     if not files:
-        return "❌ Error: No files selected", "", "❌ **Processing failed**"
+        return "❌ Error: No files selected", ""
     
     if len(files) > 10:
-        return "❌ Error: Maximum 10 files allowed at a time", "", "❌ **Processing failed**"
+        return "❌ Error: Maximum 10 files allowed at a time", ""
     
     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)
+
         # Process each file with its individual metadata
         for i, file_path in enumerate(files):
             try:
@@ -212,7 +200,7 @@ def process_files_with_progress(files, *metadata_inputs, progress=gr.Progress())
 
                 # Upload to Pinecone Assistant
                 progress((i / total_files), desc=f"☁️ Uploading {filename} to Pinecone...")
-                response = pinecone_assistant.upload_file(
+                response = assistant.upload_file(
                     file_path=destination_path, 
                     metadata=metadata, 
                     timeout=None
@@ -417,7 +405,7 @@ with gr.Blocks(
         # Tab 2: View Uploaded Files
         with gr.TabItem("📋 View Uploaded Files", id="view_tab"):
             gr.Markdown("### 📋 **Uploaded Files Management**")
-            gr.Markdown(f"Viewing files in Pinecone Assistant: **`{PINE_ASSISTANT_NAME}`**")
+            gr.Markdown("View all files currently uploaded to the Pinecone Assistant with their metadata and timestamps.")
             
             with gr.Row():
                 refresh_btn = gr.Button(