Update app.py

app.py

@@ -13,6 +13,7 @@ import shutil
 import traceback
 import json
 from typing import List, Tuple, Optional, Dict
+import requests  # Added for downloading from URLs
 
 # Clean up any existing temp files on startup to save space
 try:
@@ -321,6 +322,27 @@ def install_packages_if_needed(packages):
     
     return len(failed_packages) == 0
 
+def download_file_from_url(url: str, temp_dir: str) -> Optional[str]:
+    """Download a file from URL to temp_dir and return local path."""
+    try:
+        filename = url.split('/')[-1].split('?')[0] or 'downloaded_file'
+        if '.' not in filename:
+            filename += '.txt'  # Default extension
+        local_path = os.path.join(temp_dir, filename)
+        
+        response = requests.get(url, stream=True, timeout=30)
+        response.raise_for_status()
+        
+        with open(local_path, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+        
+        print(f"Downloaded {url} to {local_path}")
+        return local_path
+    except Exception as e:
+        print(f"Failed to download {url}: {e}")
+        return None
+
 def generate_code_with_openrouter(instruction, file_paths, previous_errors=None, attempt=1):
     """Generate Python code using OpenRouter API with error awareness."""
     
@@ -555,13 +577,31 @@ def execute_code_with_retry(code: str, max_attempts: int = 3) -> Tuple[bool, str
     
     return False, "Max attempts reached", None
 
-def process_request(instruction, files):
-    """Main processing function with self-correction"""
+def process_request(instruction, files, urls_input):
+    """Main processing function with self-correction, supporting URLs."""
     try:
         if not instruction.strip():
-            return "لطفاً دستور را وارد کنید. (فایل‌ها اختیاری هستند)", None
+            return "لطفاً دستور را وارد کنید. (فایل‌ها و لینک‌ها اختیاری هستند)", None
+        
+        file_paths = []
         
-        file_paths = [f.name for f in files] if files else []
+        # Handle uploaded files
+        if files:
+            file_paths = [f.name for f in files]
+        
+        # Handle URLs: download to temp dir
+        if urls_input and urls_input.strip():
+            temp_dir_for_downloads = tempfile.mkdtemp(prefix='url_downloads_')
+            urls = [url.strip() for url in urls_input.split(',') if url.strip()]
+            downloaded_paths = []
+            for url in urls:
+                local_path = download_file_from_url(url, temp_dir_for_downloads)
+                if local_path:
+                    downloaded_paths.append(local_path)
+            file_paths.extend(downloaded_paths)
+            print(f"Downloaded {len(downloaded_paths)} files from URLs")
+        
+        # Clean up note: Temp dirs will be cleaned on shutdown or manually if needed
         
         # Track errors for learning
         previous_errors = []
@@ -640,6 +680,8 @@ with gr.Blocks(title="AI File Processor - Self Correcting") as demo:
     - "یک فایل اکسل با 1000 نام و شماره تلفن ایرانی بساز"
     - "پس‌زمینه این تصویر را حذف کن"
     - "این فایل CSV را به JSON تبدیل کن"
+    
+    **نکته:** می‌توانید فایل‌ها را آپلود کنید یا لینک‌های فایل را (جدا شده با کاما) وارد کنید.
     """)
     
     with gr.Row():
@@ -649,7 +691,14 @@ with gr.Blocks(title="AI File Processor - Self Correcting") as demo:
             placeholder="مثال: یک نمودار دایره‌ای از داده‌های فروش بکش",
             rtl=True
         )
-        files = gr.File(file_count="multiple", label="فایل‌ها (اختیاری)")
+    
+    with gr.Row():
+        files = gr.File(file_count="multiple", label="فایل‌های آپلود شده (اختیاری)")
+        urls = gr.Textbox(
+            label="لینک فایل‌ها (جدا با کاما، اختیاری)", 
+            placeholder="https://example.com/file1.csv, https://example.com/file2.jpg",
+            info="لینک‌ها دانلود خواهند شد و حجم فایل چک نمی‌شود."
+        )
     
     btn = gr.Button("🚀 اجرا", variant="primary")
     
@@ -660,14 +709,15 @@ with gr.Blocks(title="AI File Processor - Self Correcting") as demo:
     # Examples
     gr.Examples(
         examples=[
-            ["یک فایل اکسل با 100 محصول فروشگاهی شامل نام، قیمت و موجودی بساز", []],
-            ["یک نمودار میله‌ای از داده‌های تصادفی رسم کن", []],
-            ["یک تصویر 500x500 پیکسل با رنگ‌های تصادفی بساز", []],
+            ["یک فایل اکسل با 100 محصول فروشگاهی شامل نام، قیمت و موجودی بساز", None, None],
+            ["یک نمودار میله‌ای از داده‌های تصادفی رسم کن", None, None],
+            ["یک تصویر 500x500 پیکسل با رنگ‌های تصادفی بساز", None, None],
+            ["این فایل را به فرمت JSON تبدیل کن", None, "https://example.com/sample.csv"],
         ],
-        inputs=[instruction, files],
+        inputs=[instruction, files, urls],
     )
     
-    btn.click(fn=process_request, inputs=[instruction, files], outputs=[output, file_out])
+    btn.click(fn=process_request, inputs=[instruction, files, urls], outputs=[output, file_out])
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)