Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
-from io import BytesIO
+from tqdm import tqdm
+import os
 
 # 设置Streamlit应用程序的标题
 st.title('文件下载器')
@@ -12,22 +13,35 @@ url = st.text_input('输入文件的URL：')
 if st.button('下载'):
     if url:
         try:
-            # 发送HTTP GET请求以下载文件
-            response = requests.get(url)
-            response.raise_for_status()  # 如果发生HTTP错误，将引发异常
-            
+            # 发送HEAD请求获取文件大小
+            response = requests.head(url, allow_redirects=True)
+            file_size = int(response.headers.get('content-length', 0))
+
             # 从URL中提取文件名
             filename = url.split('/')[-1]
+
+            # 下载文件并显示进度条
+            response = requests.get(url, stream=True)
+            response.raise_for_status()  # 如果发生HTTP错误，将引发异常
+
+            progress_bar = st.progress(0)
+            downloaded_size = 0
+            chunk_size = 1024  # 每次读取1KB
+
+            with open(filename, 'wb') as file:
+                for data in tqdm(response.iter_content(chunk_size=chunk_size), 
+                                 total=file_size // chunk_size, unit='KB'):
+                    file.write(data)
+                    downloaded_size += len(data)
+                    progress_bar.progress(downloaded_size / file_size)
             
-            # 创建一个字节流对象
-            file = BytesIO(response.content)
-            
-            # 提供文件下载
-            st.download_button(label='下载文件',
-                               data=file,
-                               file_name=filename,
-                               mime='application/octet-stream')
-            st.success(f"文件已准备好下载：{filename}")
+            st.success(f"文件已下载完成：{filename}")
+            with open(filename, "rb") as file:
+                st.download_button(label='下载文件',
+                                   data=file,
+                                   file_name=filename,
+                                   mime='application/octet-stream')
+
         except requests.exceptions.RequestException as e:
             st.error(f"下载文件时出错：{e}")
     else: