Update app.py

app.py

@@ -516,6 +516,13 @@ import zipfile
 import tempfile
 import shutil
 
+def find_tif_dir(root_dir):
+    """递归查找第一个包含 .tif 文件的目录"""
+    for dirpath, _, filenames in os.walk(root_dir):
+        if any(f.lower().endswith('.tif') for f in filenames):
+            return dirpath
+    return None
+
 def track_video_handler(zip_file_obj):
     """支持 ZIP 压缩包上传的 Tracking 处理函数"""
     if zip_file_obj is None:
@@ -533,11 +540,16 @@ def track_video_handler(zip_file_obj):
         with zipfile.ZipFile(zip_file_obj.name, 'r') as zip_ref:
             zip_ref.extractall(temp_dir)
 
-        print(f"🎬 Tracking - Video frame folder: {temp_dir}")
+        # 自动查找含 .tif 的子目录
+        tif_dir = find_tif_dir(temp_dir)
+        if tif_dir is None:
+            return None, f"❌ 跟踪失败: 解压后未找到任何 .tif 图像"
         
+        print(f"🎬 Tracking - Found .tif in folder: {tif_dir}")
+
         result = run_track(
             TRACK_MODEL,
-            video_dir=temp_dir,   # 传入解压目录
+            video_dir=tif_dir,   # 使用真正含图像的目录
             box=None,
             device=TRACK_DEVICE,
             output_dir="tracked_results"
@@ -551,17 +563,17 @@ def track_video_handler(zip_file_obj):
         
         result_text = f"""✅ 跟踪完成!
 
-        🎯 跟踪轨迹数量: {num_tracks}
-        📁 结果保存在: {output_dir}
-        
-        包含的文件:
-        - res_track.txt (CTC格式轨迹)
-        - 其他跟踪数据文件
-        """
+🎯 跟踪轨迹数量: {num_tracks}
+📁 结果保存在: {output_dir}
+
+包含的文件:
+- res_track.txt (CTC格式轨迹)
+- 其他跟踪数据文件
+"""
 
         print(f"✅ Tracking done - {num_tracks} tracks")
 
-        return "", result_text
+        return None, result_text
 
     except zipfile.BadZipFile:
         return None, "❌ 上传的文件不是有效的 ZIP 压缩包"
@@ -570,7 +582,6 @@ def track_video_handler(zip_file_obj):
         import traceback
         traceback.print_exc()
         return None, f"❌ 跟踪失败: {str(e)}"
-
 # ===== Gradio UI =====
 with gr.Blocks(title="Microscopy Analysis Suite", theme=gr.themes.Soft()) as demo:
     gr.Markdown(