File size: 2,154 Bytes
1e1c026 a7b0e02 1e1c026 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | import gradio as gr
import os
from zipfile import ZipFile
years = ["1985", "1986", "1987", "1988", "1989", "1990", "1991", "1992", "1993", "1994", "1995", "1996", "1997", "1998", "1999", "2000", "2001", "2002", "2003", "2004", "2005", "2006", "2007", "2008", "2009", "2010", "2011", "2012", "2013", "2014", "2015", "2016", "2017", "2018", "2019", "2020", "2021"]
months = ["all", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12"]
files_path = os.path.join("/data", "all_judgement")
def zip_csv_files(directory_path, zip_filename):
csv_files = [file for file in os.listdir(directory_path) if file.endswith('.csv')]
with ZipFile(zip_filename, 'w') as zipf:
for csv_file in csv_files:
file_path = os.path.join(directory_path, csv_file)
zipf.write(file_path, csv_file)
def get_file(year_index, month_index):
file_path = os.path.join(files_path, years[year_index])
name = str(years[year_index])+'年'+ str(months[month_index] + "月")
text = "File not exit! Change the path plz, sister.Meng: "
if months[month_index] == "all":
directory_to_zip = file_path
zip_file_name = str(years[year_index])+'年'+ "all.zip"
zip_csv_files(directory_to_zip, zip_file_name)
name = zip_file_name
text = "都压缩起来了奥，文件可能嘎嘎大！稍微等会萌子姐: " + name
return text, zip_file_name
else:
file_path = os.path.join(file_path, name+".csv")
if not os.path.exists(file_path):
text = text + name
return text, os.path.join("/data", "all_judgement", "readme.md")
else:
text = "马上就好奥萌子姐: " + name
return text, file_path
output_component = gr.File()
input_component = [
gr.Dropdown(years, type="index", label="years", value="2021"),
gr.Dropdown(months, type="index", label="months", value="all")]
demo = gr.Interface(fn=get_file, inputs = input_component, outputs = ["label", output_component], cache_examples=True, title="输入年月查询")
demo.launch(share=True, server_name='0.0.0.0', server_port=8043, show_api=True)
|