Update app.py

app.py

@@ -1,18 +1,12 @@
-import json
-import os
-from pathlib import Path
 import re
 import threading
 import time
-from tkinter import filedialog
-import traceback
+from pathlib import Path
 import gradio as gr
 import requests
 from tqdm import tqdm
 import tempfile
 
-lock = threading.Lock()
-event = threading.Event()
 requests.packages.urllib3.disable_warnings()
 
 reference_subject = {
@@ -143,13 +137,12 @@ def to_file(file, source_type, name=""):
     result["type"] = source_type
     return result
 
-# 修改点1：添加uid参数
 def analyze_homework(homework, include_text, uid):  
     hwId = homework["hwId"]
     hwType = homework["hwType"]
     stuHwId = homework["stuHwId"]
     file_list = []
-    data = {"base": {"appId": "APP"}, "params": {"hwId": hwId, "stuHwId": stuHwId, "studentId": uid}}  # 使用传入的uid值
+    data = {"base": {"appId": "APP"}, "params": {"hwId": hwId, "stuHwId": stuHwId, "studentId": uid}}
     
     if hwType == 102:
         response = post("https://mhw.zhixue.com/hwreport/question/getStuReportDetail", data)
@@ -240,7 +233,6 @@ def query_homework(uid, tlsysSessionId, subject, status, max_count):
         homework_list = homework_list_temp
         return token, gr.update(choices=homework_list, value=[])
 
-# 修改点2：添加uid参数
 def parse_homework(token, include_text, homework_selection, uid):  
     result = []
     for i in homework_selection:
@@ -250,8 +242,7 @@ def parse_homework(token, include_text, homework_selection, uid):
     selected_homework = result
     file_list = []
     for i in tqdm(selected_homework, unit=""):
-        # 修改点3：传递uid参数
-        file_list += analyze_homework(homework_list_oringin[i], include_text, uid)  
+        file_list += analyze_homework(homework_list_oringin[i], include_text, uid)
         global file_list_output
         file_list_output = [file["name"] for file in file_list]
         global homework_downloaded_path
@@ -259,7 +250,8 @@ def parse_homework(token, include_text, homework_selection, uid):
         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):
+
+def download_file(homework_view):
     download_list = []
     for i in homework_view:
         for j in range(len(file_list_output)):
@@ -273,12 +265,37 @@ def download_file(token, homework_view):
                     download_list.append(temp_path)
     return download_list
 
+def save_config(uid, tlsysSessionId):
+    config_content = f"uid={uid}\ntlsysSessionId={tlsysSessionId}"
+    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
+        f.write(config_content)
+        return f.name
+
+def load_config(file):
+    if not file:
+        return None, None
+    with open(file.name, "r") as f:
+        content = f.read()
+    uid = re.search(r"uid=(\S+)", content)
+    tlsysSessionId = re.search(r"tlsysSessionId=(\S+)", content)
+    return (uid.group(1) if uid else None, (tlsysSessionId.group(1) if tlsysSessionId else None))
+
 with gr.Blocks(title="智学网作业获取器") as demo:
     gr.Markdown("# 🚀 智学网作业获取器")
     gr.Markdown("## Backfront Created by Levrium,UI Design by Start_ten")
     gr.Markdown("操作说明请见https://zhuanlan.zhihu.com/p/691808543")
+    
+    # 新增配置文件区域
+    with gr.Row():
+        with gr.Column(scale=2):
+            config_upload = gr.File(label="上传配置文件", type="filepath", file_types=[".txt"])
+        with gr.Column(scale=1):
+            load_config_btn = gr.Button("识别配置文件", variant="secondary")
+            save_config_btn = gr.Button("导出配置文件", variant="secondary")
+        with gr.Column(scale=2):
+            config_download = gr.File(label="下载配置文件", interactive=False)
+    
     with gr.Row():
-
         with gr.Column():
             gr.Markdown("### 📝 查询作业")
             uid = gr.Textbox(
@@ -359,20 +376,32 @@ with gr.Blocks(title="智学网作业获取器") as demo:
                 interactive=True
             )
             submit_btn = gr.Button("解析作业", variant="primary")
-            # 修改点4：添加uid输入
             submit_btn.click(
                 fn=parse_homework,
-                inputs=[token, text_parse, homework_selection, uid],  # 添加uid输入
-                outputs=[homework_view, ]
+                inputs=[token, text_parse, homework_selection, uid],
+                outputs=[homework_view]
             )
         with gr.Column():
             file_output = gr.File(label="作业文件", interactive=False)
             download_btn = gr.Button("下载作业", variant="primary")
             download_btn.click(
                 fn=download_file,
-                inputs=[token, homework_view],
+                inputs=[homework_view],
                 outputs=[file_output]
             )
+    
+    # 配置文件功能绑定
+    load_config_btn.click(
+        fn=load_config,
+        inputs=[config_upload],
+        outputs=[uid, tlsysSessionId]
+    )
+    
+    save_config_btn.click(
+        fn=save_config,
+        inputs=[uid, tlsysSessionId],
+        outputs=[config_download]
+    )
 
 if __name__ == "__main__":
     demo.launch()