Spaces:
Sleeping
Sleeping
Error handling and little update
Browse files
app.py
CHANGED
@@ -1,21 +1,29 @@
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
import os
|
|
|
4 |
|
5 |
def download_file(url):
|
6 |
"""Downloads a file from a URL and returns the local file path."""
|
|
|
|
|
|
|
7 |
try:
|
8 |
response = requests.get(url, stream=True)
|
9 |
response.raise_for_status() # Raise an exception for bad status codes
|
10 |
|
11 |
-
# Generate a unique temporary filename
|
12 |
-
|
|
|
|
|
|
|
|
|
13 |
with open(temp_filename, 'wb') as f:
|
14 |
-
for chunk in response.iter_content(chunk_size=
|
15 |
f.write(chunk)
|
16 |
return temp_filename
|
17 |
except requests.exceptions.MissingSchema:
|
18 |
-
return "Error: Invalid URL format.
|
19 |
except requests.exceptions.ConnectionError:
|
20 |
return "Error: Could not connect to the server. Please check your internet connection."
|
21 |
except requests.exceptions.RequestException as e:
|
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
import os
|
4 |
+
import re
|
5 |
|
6 |
def download_file(url):
|
7 |
"""Downloads a file from a URL and returns the local file path."""
|
8 |
+
if not url.startswith("http://") and not url.startswith("https://"):
|
9 |
+
url = "http://" + url # Prepend "http://" if not present
|
10 |
+
|
11 |
try:
|
12 |
response = requests.get(url, stream=True)
|
13 |
response.raise_for_status() # Raise an exception for bad status codes
|
14 |
|
15 |
+
# Generate a safe and unique temporary filename
|
16 |
+
original_filename = os.path.basename(url)
|
17 |
+
# Remove invalid characters from filename
|
18 |
+
safe_filename = re.sub(r'[^\w\-_\. ]', '_', original_filename)
|
19 |
+
temp_filename = f"{safe_filename}"
|
20 |
+
|
21 |
with open(temp_filename, 'wb') as f:
|
22 |
+
for chunk in response.iter_content(chunk_size=8192000):
|
23 |
f.write(chunk)
|
24 |
return temp_filename
|
25 |
except requests.exceptions.MissingSchema:
|
26 |
+
return "Error: Invalid URL format. Even after adding 'http://', the URL is still invalid."
|
27 |
except requests.exceptions.ConnectionError:
|
28 |
return "Error: Could not connect to the server. Please check your internet connection."
|
29 |
except requests.exceptions.RequestException as e:
|