Update app.py

app.py

@@ -7,7 +7,7 @@ import mimetypes
 import zipfile
 import tempfile
 from datetime import datetime
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Union, Any
 from pathlib import Path
 import requests
 import validators
@@ -94,7 +94,7 @@ class URLProcessor:
         try:
             file_id = re.search(r'/file/d/([a-zA-Z0-9_-]+)', url)
             if not file_id:
-                logger.error(f"Invalid Google Drive URL: { url}")
+                logger.error(f"Invalid Google Drive URL: {url}")
                 return None
                 
             direct_url = f"https://drive.google.com/uc?export=download&id={file_id.group(1)}"
@@ -165,7 +165,7 @@ class FileProcessor:
         except Exception:
             return False
 
-    def process_files(self, files) -> List[Dict]:
+    def process_files(self, files: Union[List[gr.File], List[str]]) -> List[Dict]:
         """Process multiple uploaded files and return a single JSON extraction"""
         if not files:
             return []
@@ -173,15 +173,17 @@ class FileProcessor:
         combined_data = []
         try:
             for file in files:
-                file_size = os.path.getsize(file.name)
+                # Check if the file is a Gradio File object or a string path
+                file_name = file.name if isinstance(file, gr.File) else file
+                file_size = os.path.getsize(file_name)
                 if file_size > self.max_file_size:
                     logger.warning(f"File size ({file_size} bytes) exceeds maximum allowed size")
                     continue  # Skip this file
 
-                if zipfile.is_zipfile(file.name):
-                    combined_data.extend(self._process_zip_file(file.name))
+                if zipfile.is_zipfile(file_name):
+                    combined_data.extend(self._process_zip_file(file_name))
                 else:
-                    combined_data.extend(self._process_single_file(file))
+                    combined_data.extend(self._process_single_file(file_name))
 
         except Exception as e:
             logger.error(f"Error processing files: {str(e)}")
@@ -213,17 +215,17 @@ class FileProcessor:
                                 logger.error(f"Error reading file {filename}: {str(e)}")
         return results
 
-    def _process_single_file(self, file) -> List[Dict]:
+    def _process_single_file(self, file_path: str) -> List[Dict]:
         try:
-            file_stat = os.stat(file.name)
-            with open(file.name, 'r', encoding='utf-8', errors='ignore') as f:
+            file_stat = os.stat(file_path)
+            with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
                 content = f.read()
             
             return [{
                 'source': 'file',
-                'filename': os.path.basename(file.name),
+                'filename': os.path.basename(file_path),
                 'file_size': file_stat.st_size,
-                'mime_type': mimetypes.guess_type(file.name)[0 ],
+                'mime_type': mimetypes.guess_type(file_path)[0],
                 'created': datetime.fromtimestamp(file_stat.st_ctime).isoformat(),
                 'modified': datetime.fromtimestamp(file_stat.st_mtime).isoformat(),
                 'content': content,
@@ -286,7 +288,8 @@ def create_interface():
         with gr.Tab("File Input"):
             file_input = gr.File(
                 label="Upload text files or ZIP archives",
-                file_types=[".txt", ".zip", ".md", ".csv", ".json", ".xml"]
+                file_types=[".txt", ".zip", ".md", ".csv", ".json", ".xml"],
+                multiple=True  # Allow multiple file uploads
             )
 
         with gr.Tab("Text Input"):
@@ -326,7 +329,7 @@ def create_interface():
                 
                 # Process URLs
                 if urls:
-                    url_list = re.split(r'[,\n]', urls)
+                    url_list = re.split(r '[,\n]', urls)
                     url_list = [url.strip() for url in url_list if url.strip()]
                     
                     for url in url_list:
@@ -344,7 +347,7 @@ def create_interface():
                 # Process files
                 if files:
                     combined_data = file_processor.process_files(files)
-                    results.extend (combined_data)
+                    results.extend(combined_data)
         
                 # Process text input
                 if text:
@@ -402,7 +405,7 @@ def create_interface():
         gr.Markdown("""
         ### Usage Guidelines
         - **URL Processing**: Enter valid HTTP/HTTPS URLs
-        - **File Input**: Upload text files or ZIP archives (single file allowed)
+        - **File Input**: Upload multiple text files or ZIP archives
         - **Text Input**: Direct text processing
         - **Chat**: Load your JSON data and ask questions about it
         - Advanced cleaning and validation included