Update app.py

app.py

@@ -66,8 +66,8 @@ def process_video(video_file, question):
     return generated_text.split("ASSISTANT: ", 1)[-1].strip()
 
 @spaces.GPU
-def analyze_videos(video_files, selected_question):
-    """Analyzes all videos with the selected question."""
+def analyze_videos(video_files, selected_questions):
+    """Analyzes all videos with the selected questions."""
     all_results = {}
 
     questions = {
@@ -79,8 +79,11 @@ def analyze_videos(video_files, selected_question):
 
     for video_file in video_files:
         video_name = os.path.basename(video_file.name)
-        answer = process_video(video_file, questions[selected_question])
-        all_results[video_name] = {selected_question: answer}
+        all_results[video_name] = {}
+        for question_key in selected_questions:
+            answer = process_video(video_file, questions[question_key])
+            # Simple True/False determination (You might want to refine this)
+            all_results[video_name][question_key] = "true" if "yes" in answer.lower() else "false"
     return json.dumps(all_results, indent=4)
 
 # Define Gradio interface 
@@ -88,12 +91,12 @@ iface = gr.Interface(
     fn=analyze_videos,
     inputs=[
         gr.File(label="Upload Videos", file_count="multiple"),
-        gr.Dropdown(["hands_free", "standing/sitting", "interaction_with_background", "indoors/outdoors"], 
-                    label="Select Question to Apply")
+        gr.CheckboxGroup(["hands_free", "standing/sitting", "interaction_with_background", "indoors/outdoors"], 
+                        label="Select Questions to Apply") 
     ],
     outputs=gr.JSON(label="Analysis Results"), 
     title="Video Analysis",
-    description="Upload videos and select a question to analyze."
+    description="Upload videos and select questions to analyze."
 )
 
 if __name__ == "__main__":