Ethscriptions commited on
Commit
8d3bc9c
·
verified ·
1 Parent(s): 9aaf9d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import streamlit as st
2
  import requests
3
  from tqdm import tqdm
4
  import os
 
5
 
6
  # 设置Streamlit应用程序的标题
7
  st.title('文件下载器')
@@ -18,7 +19,12 @@ if st.button('下载'):
18
  file_size = int(response.headers.get('content-length', 0))
19
 
20
  # 从URL中提取文件名
21
- filename = url.split('/')[-1]
 
 
 
 
 
22
 
23
  # 下载文件并显示进度条
24
  response = requests.get(url, stream=True)
@@ -44,5 +50,7 @@ if st.button('下载'):
44
 
45
  except requests.exceptions.RequestException as e:
46
  st.error(f"下载文件时出错:{e}")
 
 
47
  else:
48
  st.warning('请输入一个有效的URL。')
 
2
  import requests
3
  from tqdm import tqdm
4
  import os
5
+ from urllib.parse import urlparse
6
 
7
  # 设置Streamlit应用程序的标题
8
  st.title('文件下载器')
 
19
  file_size = int(response.headers.get('content-length', 0))
20
 
21
  # 从URL中提取文件名
22
+ parsed_url = urlparse(url)
23
+ filename = os.path.basename(parsed_url.path)
24
+ if not filename:
25
+ filename = 'downloaded_file'
26
+ if len(filename) > 255:
27
+ filename = filename[:255]
28
 
29
  # 下载文件并显示进度条
30
  response = requests.get(url, stream=True)
 
50
 
51
  except requests.exceptions.RequestException as e:
52
  st.error(f"下载文件时出错:{e}")
53
+ except Exception as e:
54
+ st.error(f"发生错误:{e}")
55
  else:
56
  st.warning('请输入一个有效的URL。')