store

app.py

@@ -171,6 +171,7 @@ def create_video_survey_app():
         # 状态变量
         current_question_idx = gr.State(0)
         current_method_mapping = gr.State({})
+        all_ratings = gr.State({})  # 存储所有场景的评分
         
         # 进度显示
         with gr.Row():
@@ -217,18 +218,21 @@ def create_video_survey_app():
                 })
         
         # 更新问题显示
-        def update_question(question_idx, save_previous=False, prev_ratings=None, prev_mapping=None):
+        def update_question(question_idx, all_ratings_dict, save_previous=False, prev_ratings=None, prev_mapping=None):
             if question_idx < 0:
                 question_idx = 0
             if question_idx >= len(question_folders):
                 question_idx = len(question_folders) - 1
             
-            # 如果需要，保存上一题的评分
+            # 如果需要，保存上一题的评分到字典
             save_msg = ""
             if save_previous and prev_ratings and prev_mapping:
-                prev_scene = question_folders[question_idx - 1] if question_idx > 0 else None
-                if prev_scene:
-                    save_msg = save_ratings(prev_scene, prev_ratings, prev_mapping)
+                prev_idx = question_idx - 1 if question_idx > 0 else question_idx
+                prev_scene = question_folders[prev_idx]
+                all_ratings_dict[prev_scene] = {
+                    'ratings': prev_ratings,
+                    'mapping': prev_mapping
+                }
             
             scene_name = question_folders[question_idx]
             videos, method_mapping = get_videos_for_question(scene_name)
@@ -242,13 +246,27 @@ def create_video_survey_app():
                 else:
                     video_updates.append(gr.update(value=None))
             
-            # 重置评分
-            rating_updates = [gr.update(value=3) for _ in range(12)]  # 4个视频 x 3个评分，每个默认值为3
+            # 检查是否有已保存的评分
+            rating_updates = []
+            if scene_name in all_ratings_dict:
+                # 恢复之前的评分
+                saved_ratings = all_ratings_dict[scene_name]['ratings']
+                for i in range(4):
+                    method_name = f"Method {chr(65+i)}"
+                    if method_name in saved_ratings:
+                        rating_updates.append(gr.update(value=saved_ratings[method_name]['dynamic_quality']))
+                        rating_updates.append(gr.update(value=saved_ratings[method_name]['static_consistency']))
+                        rating_updates.append(gr.update(value=saved_ratings[method_name]['overall_quality']))
+                    else:
+                        rating_updates.extend([gr.update(value=3), gr.update(value=3), gr.update(value=3)])
+            else:
+                # 重置为默认评分
+                rating_updates = [gr.update(value=3) for _ in range(12)]  # 4个视频 x 3个评分，每个默认值为3
             
             question_markdown = f"**场景 {question_idx + 1} / {len(question_folders)}**: `{scene_name}`"
             
             return (
-                [question_idx, method_mapping, question_markdown, save_msg] +
+                [question_idx, method_mapping, all_ratings_dict, question_markdown, save_msg] +
                 video_updates +
                 rating_updates
             )
@@ -267,32 +285,48 @@ def create_video_survey_app():
             return ratings
         
         # 下一题按钮
-        def on_next(question_idx, method_mapping, *rating_values):
+        def on_next(question_idx, method_mapping, all_ratings_dict, *rating_values):
             # 收集当前评分
             current_ratings = collect_ratings(*rating_values)
             
-            # 保存当前评分
+            # 保存当前评分到字典
             scene_name = question_folders[question_idx]
-            save_msg = save_ratings(scene_name, current_ratings, method_mapping)
+            all_ratings_dict[scene_name] = {
+                'ratings': current_ratings,
+                'mapping': method_mapping
+            }
             
             # 移动到下一题
             new_idx = question_idx + 1
             if new_idx >= len(question_folders):
+                # 所有题目完成，保存到文件
+                for scene, data in all_ratings_dict.items():
+                    save_ratings(scene, data['ratings'], data['mapping'])
+                
                 return [
                     question_idx,
                     method_mapping,
+                    all_ratings_dict,
                     f"**✅ 所有场景已完成！/ All scenes completed!**",
-                    save_msg + "\n🎉 感谢参与！/ Thank you for participating!"
+                    "🎉 感谢参与！所有评分已保存 / Thank you! All ratings saved"
                 ] + [gr.update(value=None)] * 4 + [gr.update(value=3)] * 12
             
-            return update_question(new_idx)
+            return update_question(new_idx, all_ratings_dict)
         
         # 上一题按钮
-        def on_prev(question_idx, *args):
+        def on_prev(question_idx, method_mapping, all_ratings_dict, *rating_values):
+            # 收集当前评分并保存
+            current_ratings = collect_ratings(*rating_values)
+            scene_name = question_folders[question_idx]
+            all_ratings_dict[scene_name] = {
+                'ratings': current_ratings,
+                'mapping': method_mapping
+            }
+            
             new_idx = question_idx - 1
             if new_idx < 0:
                 new_idx = 0
-            return update_question(new_idx)
+            return update_question(new_idx, all_ratings_dict)
         
         # 收集所有评分组件
         all_rating_inputs = []
@@ -302,10 +336,11 @@ def create_video_survey_app():
         # 绑定事件
         next_btn.click(
             on_next,
-            inputs=[current_question_idx, current_method_mapping] + all_rating_inputs,
+            inputs=[current_question_idx, current_method_mapping, all_ratings] + all_rating_inputs,
             outputs=[
                 current_question_idx,
                 current_method_mapping,
+                all_ratings,
                 question_text,
                 status_text
             ] + video_components + all_rating_inputs
@@ -313,10 +348,11 @@ def create_video_survey_app():
         
         prev_btn.click(
             on_prev,
-            inputs=[current_question_idx] + all_rating_inputs,
+            inputs=[current_question_idx, current_method_mapping, all_ratings] + all_rating_inputs,
             outputs=[
                 current_question_idx,
                 current_method_mapping,
+                all_ratings,
                 question_text,
                 status_text
             ] + video_components + all_rating_inputs
@@ -324,10 +360,11 @@ def create_video_survey_app():
         
         # 初始化第一个问题
         demo.load(
-            lambda: update_question(0),
+            lambda: update_question(0, {}),
             outputs=[
                 current_question_idx,
                 current_method_mapping,
+                all_ratings,
                 question_text,
                 status_text
             ] + video_components + all_rating_inputs