Update app.py

app.py

@@ -9,6 +9,7 @@ import traceback
 import gradio as gr
 import requests
 from tqdm import tqdm
+import tempfile
 
 lock = threading.Lock()
 event = threading.Event()
@@ -189,7 +190,7 @@ def query_homework(uid, tlsysSessionId, subject, status, max_count):
     headers[0].update({"Cookie": f"tlsysSessionId={tlsysSessionId}"})
     successful = get_token()
     if not successful:
-        return  "","","获取 token 失败"
+        return  "获取 token 失败",""
     page_size = max_count
     if subject == ["语文", "数学", "英语", "物理", "化学", "生物", "地理", "历史", "政治",'通用技术', '信息技术']:
         subjects = ["-1"]
@@ -255,13 +256,21 @@ def parse_homework(token, include_text, homework_selection, uid):
         file_list_output = [file["name"] for file in file_list]
         global homework_downloaded_path
         homework_downloaded_path = [file["path"] for file in file_list]
+        global homework_is_text
+        homework_is_text = [file["is_text"] for file in file_list]
     return gr.update(choices=file_list_output, value=[])
 def download_file(token, homework_view):
     download_list = []
     for i in homework_view:
         for j in range(len(file_list_output)):
             if i == file_list_output[j]:
-                download_list.append(homework_downloaded_path[j])
+                if not homework_is_text[j]:
+                    download_list.append(homework_downloaded_path[j])
+                else:
+                    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
+                        f.write(homework_downloaded_path[j])
+                        temp_path = f.name
+                    download_list.append(temp_path)
     return download_list
 
 with gr.Blocks(title="智学网作业获取器") as demo: