Update app.py

app.py

@@ -27,6 +27,7 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
     global task_counter
     task_counter += 1
     task_id = task_counter
+    print_log(task_id, input2,  input1, "input1")
     log = "转换过程非常慢，请耐心等待。显示文件列表不代表转换完成。如果未发生错误，转换结束会显示”任务完成“\n"
     yield [], log
     if input2 == None or input2.strip() == "":
@@ -95,7 +96,8 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
                             yield [], log
                             return
         elif input1 is not None:
-            file_path = input1.name
+            print("check file" , input1, os.path.exists(input1))
+            file_path = input1
             log += f"使用上传的文件: {file_path}\n"
             print_log(task_id, input2, "使用上传文件", "开始")
             yield [], log
@@ -105,6 +107,20 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
             yield [], log
             return
 
+        # 检查文件大小
+        try:
+            file_size = os.path.getsize(file_path) / 1024 /1024  # 转换为 KB
+            if file_size > 100 :
+                log += f"文件太大，建议 100MB 以内，当前文件大小为 {file_size } MB。\n"
+                print_log(task_id, input2, "文件太大（"+ file_size +"MB）", "失败")
+                yield [], log
+                return
+        except Exception as e:
+            log += f"获取文件大小失败: {str(e)}\n"
+            print_log(task_id, input2, "检查文件大小", f"失败: {str(e)}")
+            yield [], log
+            return
+
         # 生成新文件夹用于暂存结果
         output_folder = os.path.join(os.getcwd(), str(uuid.uuid4()))
         os.makedirs(output_folder, exist_ok=True)
@@ -199,9 +215,11 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
                 if output == '' and process.poll() is not None:
                     break
                 if output:
-                    # log += output.strip() + '\n'
+                    # 
                     if log_to_terminal:
                         print(output.strip())
+                        log += output.strip() + '\n'
+                        yield [], log
             returncode = process.poll()
             if returncode != 0:
                 log += f"执行命令: {command} 失败，返回码: {returncode}\n"