Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import os
 import re
+import mimetypes
 
 def download_file(url):
   """Downloads a file from a URL and returns the local file path."""
@@ -18,6 +19,12 @@ def download_file(url):
     safe_filename = re.sub(r'[^\w\-_\. ]', '_', original_filename) 
     temp_filename = f"{safe_filename}"
 
+    # Infer file extension from content type
+    content_type = response.headers['content-type']
+    ext = mimetypes.guess_extension(content_type)
+    if ext and not temp_filename.endswith(ext):  # Append extension if not already present
+        temp_filename += ext
+
     with open(temp_filename, 'wb') as f:
       for chunk in response.iter_content(chunk_size=8192000):
         f.write(chunk)
@@ -35,6 +42,7 @@ iface = gr.Interface(
     outputs=gr.File(),
     title="File Downloader",
     description="Enter the URL of an image, video, document, etc. to download it.",
+    concurrency_limit=None
 )
 
 iface.launch()