store

app.py

@@ -166,208 +166,110 @@ def create_video_survey_app():
           - **静态一致性** / Static Consistency: 视频中静态物体的稳定性和一致性
           - **整体质量** / Overall Quality: 视频的整体观感
         - 评分范围：1-5分（5分最好）/ Rating scale: 1-5 (5 = Best)
+        - 完成所有评分后，点击底部的"提交所有评分"按钮 / After rating all videos, click "Submit All Ratings" at the bottom
         """)
         
-        # 状态变量
-        current_question_idx = gr.State(0)
-        current_method_mapping = gr.State({})
-        all_ratings = gr.State({})  # 存储所有场景的评分
+        # 存储所有评分组件
+        all_scene_components = []
         
-        # 进度显示
-        with gr.Row():
-            prev_btn = gr.Button("⬅️ 上一题 / Previous", size="sm")
-            question_text = gr.Markdown(f"**场景 1 / {len(question_folders)}**")
-            next_btn = gr.Button("下一题 / Next ➡️", size="sm", variant="primary")
-        
-        status_text = gr.Textbox(label="状态 / Status", interactive=False, visible=False)
-        
-        # 视频显示区域（4个视频）
-        video_components = []
-        rating_components = []
-        
-        for i in range(4):
-            method_name = f"Method {chr(65+i)}"
+        # 为每个场景创建界面
+        for scene_idx, scene_name in enumerate(question_folders):
+            videos, method_mapping = get_videos_for_question(scene_name)
             
             with gr.Group():
-                gr.Markdown(f"### 🎥 {method_name}")
-                
-                video = gr.Video(label="", height=300)
-                video_components.append(video)
+                gr.Markdown(f"## 场景 {scene_idx + 1} / Scene {scene_idx + 1}: `{scene_name}`")
                 
+                # 4个视频横向排列
                 with gr.Row():
-                    dynamic = gr.Slider(
-                        minimum=1, maximum=5, step=1, value=3,
-                        label="动态质量 / Dynamic Quality",
-                        info="1=差 / Poor, 5=优秀 / Excellent"
-                    )
-                    static = gr.Slider(
-                        minimum=1, maximum=5, step=1, value=3,
-                        label="静态一致性 / Static Consistency",
-                        info="1=差 / Poor, 5=优秀 / Excellent"
-                    )
-                    overall = gr.Slider(
-                        minimum=1, maximum=5, step=1, value=3,
-                        label="整体质量 / Overall Quality",
-                        info="1=差 / Poor, 5=优秀 / Excellent"
-                    )
+                    scene_ratings = []
+                    for i in range(4):
+                        method_name = f"Method {chr(65+i)}"
+                        
+                        with gr.Column():
+                            gr.Markdown(f"### 🎥 {method_name}")
+                            
+                            # 视频
+                            if method_name in videos:
+                                video = gr.Video(value=videos[method_name], label="", height=300)
+                            else:
+                                video = gr.Video(value=None, label="", height=300, visible=False)
+                            
+                            # 三个评分滑块（垂直排列）
+                            dynamic = gr.Slider(
+                                minimum=1, maximum=5, step=1, value=3,
+                                label="动态质量 / Dynamic",
+                                info="1=差, 5=优秀"
+                            )
+                            static = gr.Slider(
+                                minimum=1, maximum=5, step=1, value=3,
+                                label="静态一致性 / Static",
+                                info="1=差, 5=优秀"
+                            )
+                            overall = gr.Slider(
+                                minimum=1, maximum=5, step=1, value=3,
+                                label="整体质量 / Overall",
+                                info="1=差, 5=优秀"
+                            )
+                            
+                            scene_ratings.append({
+                                "method": method_name,
+                                "dynamic": dynamic,
+                                "static": static,
+                                "overall": overall
+                            })
                 
-                rating_components.append({
-                    "dynamic": dynamic,
-                    "static": static,
-                    "overall": overall
+                all_scene_components.append({
+                    "scene_name": scene_name,
+                    "method_mapping": method_mapping,
+                    "ratings": scene_ratings
                 })
         
-        # 更新问题显示
-        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_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)
-            
-            # 更新视频显示
-            video_updates = []
-            for i in range(4):
-                method_name = f"Method {chr(65+i)}"
-                if method_name in videos:
-                    video_updates.append(gr.update(value=videos[method_name]))
-                else:
-                    video_updates.append(gr.update(value=None))
-            
-            # 检查是否有已保存的评分
-            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, all_ratings_dict, question_markdown, save_msg] +
-                video_updates +
-                rating_updates
-            )
-        
-        # 收集当前评分
-        def collect_ratings(*rating_values):
-            ratings = {}
-            for i in range(4):
-                method_name = f"Method {chr(65+i)}"
-                base_idx = i * 3
-                ratings[method_name] = {
-                    "dynamic_quality": rating_values[base_idx],
-                    "static_consistency": rating_values[base_idx + 1],
-                    "overall_quality": rating_values[base_idx + 2]
-                }
-            return ratings
+        # 提交按钮
+        gr.Markdown("---")
+        submit_btn = gr.Button("📤 提交所有评分 / Submit All Ratings", variant="primary", size="lg")
+        status_output = gr.Markdown("")
         
-        # 下一题按钮
-        def on_next(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 >= len(question_folders):
-                # 所有题目完成，保存到文件
-                for scene, data in all_ratings_dict.items():
-                    save_ratings(scene, data['ratings'], data['mapping'])
+        # 提交函数
+        def submit_all_ratings(*all_rating_values):
+            try:
+                # 解析所有评分
+                value_idx = 0
+                for scene_data in all_scene_components:
+                    scene_name = scene_data["scene_name"]
+                    method_mapping = scene_data["method_mapping"]
+                    
+                    ratings = {}
+                    for rating_comp in scene_data["ratings"]:
+                        method_name = rating_comp["method"]
+                        ratings[method_name] = {
+                            "dynamic_quality": all_rating_values[value_idx],
+                            "static_consistency": all_rating_values[value_idx + 1],
+                            "overall_quality": all_rating_values[value_idx + 2]
+                        }
+                        value_idx += 3
+                    
+                    # 保存评分
+                    save_ratings(scene_name, ratings, method_mapping)
                 
-                return [
-                    question_idx,
-                    method_mapping,
-                    all_ratings_dict,
-                    f"**✅ 所有场景已完成！/ All scenes completed!**",
-                    "🎉 感谢参与！所有评分已保存 / Thank you! All ratings saved"
-                ] + [gr.update(value=None)] * 4 + [gr.update(value=3)] * 12
-            
-            return update_question(new_idx, all_ratings_dict)
-        
-        # 上一题按钮
-        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, all_ratings_dict)
+                return "✅ **所有评分已成功保存！感谢参与！** / All ratings saved successfully! Thank you for participating!"
+            except Exception as e:
+                return f"❌ **保存失败** / Save failed: {str(e)}"
         
-        # 收集所有评分组件
+        # 收集所有评分输入
         all_rating_inputs = []
-        for comp in rating_components:
-            all_rating_inputs.extend([comp["dynamic"], comp["static"], comp["overall"]])
-        
-        # 绑定事件
-        next_btn.click(
-            on_next,
-            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
-        )
-        
-        prev_btn.click(
-            on_prev,
-            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
-        )
+        for scene_data in all_scene_components:
+            for rating_comp in scene_data["ratings"]:
+                all_rating_inputs.extend([
+                    rating_comp["dynamic"],
+                    rating_comp["static"],
+                    rating_comp["overall"]
+                ])
         
-        # 初始化第一个问题
-        demo.load(
-            lambda: update_question(0, {}),
-            outputs=[
-                current_question_idx,
-                current_method_mapping,
-                all_ratings,
-                question_text,
-                status_text
-            ] + video_components + all_rating_inputs
+        # 绑定提交按钮
+        submit_btn.click(
+            submit_all_ratings,
+            inputs=all_rating_inputs,
+            outputs=status_output
         )
     
     return demo