Update app.py

app.py

@@ -6,13 +6,13 @@ from fastbook import *
 import os
 from zipfile import ZipFile
 import shutil
+import numpy as np
 
 
 title="批量下载图片"
 description="""
     1. 可以下载不同种类的图片，图片名称使用空格隔开，比如： 张学友 刘德华 黎明 郭富城\n
     2. 下载的数量应该大于0，程序运行的速度与下载的图片数量和种类个数有关\n
-    3. 图片存储地址不能包含空格，有效的存储地址：E:\图片
 """
 article="<p style='text-align: center'><center><img src='https://visitor-badge.glitch.me/badge?page_id=kingabzpro/Rick_and_Morty_Bot' alt='visitor badge'></center></p>"
 
@@ -25,15 +25,18 @@ def zipDir(dirpath,outfullname):
     zipp.close()
     return outfullname
 
-def download_image(name,number,overwrite,dirs):
-    dirs=dirs.split(' ')
-    dirs=[i for i in dirs if i != '']
-    if len(dirs)==0:
-        return f"请输入有效的存储地址"
-    else:
-        dirs=''.join(dirs)
-    if not os.path.exists(dirs):  # 如果不存在，则创建文件夹。与代码文件所在目录相同。如果存在就直接跳过整段代码
-        os.makedirs(dirs)       # 创建文件夹
+def download_image(name,number,overwrite):
+    #dirs=dirs.split(' ')
+    #dirs=[i for i in dirs if i != '']
+    #if len(dirs)==0:
+    #    return f"请输入有效的存储地址"
+   # else:
+     #   dirs=''.join(dirs)
+    #if not os.path.exists(dirs):  # 如果不存在，则创建文件夹。与代码文件所在目录相同。如果存在就直接跳过整段代码
+       # os.makedirs(dirs)       # 创建文件夹
+       
+    # 因为在使用Huggingface的时候，其不能访问本地电脑，所以直接内置一个地址
+    dirs='E:\图片'
     names=name.split(' ')
     names=[i for i in names if i != '']
     if len(names)==0:
@@ -59,7 +62,8 @@ def download_image(name,number,overwrite,dirs):
         # dest 下载目录
         # urls 链接
         download_images(dest,urls=results,preserve_filename=ow)
-        
+    
+    # 统计图片信息    
     files1=os.listdir(dirs)
     files2=[]
     num1=len(files1)
@@ -75,7 +79,10 @@ def download_image(name,number,overwrite,dirs):
         printing.append(pp)
     prt='\n'.join(printing)
     
-    out1=f'图片存放地址{dirs}\n其中：\n{prt}\n'
+    num2=np.array(num2)
+    
+    
+    out1=f'一共下载{num2.sum()}张图片\n其中：\n{prt}\n'
     
     out2=zipDir(dirs,'pictures.zip')
     # 删除文件夹
@@ -97,10 +104,12 @@ demo=gr.Interface(
         ),
         gr.Number(value=10, label="下载图片数量，比如：10"),
         gr.Radio(['不允许重复下载','允许重复下载'],label='是否允许重复下载'),
-        gr.Textbox(
-            label="图片存储地址，比如： E:\图片",
-            lines=1,
-            placeholder='E:\图片',
+        
+        # 不用输入下载地址了，也就不需要这个了
+        #gr.Textbox(
+         #   label="图片存储地址，比如： E:\图片",
+          #  lines=1,
+           # placeholder='E:\图片',
             #value="E:\图片",
             
         ),