Update app.py

app.py

@@ -39,20 +39,26 @@ def process_video_and_questions(video, standing, hands, location, screen):
     response = describe_video(video, final_prompt)
     final_response = f"<video_name>{video_name}</video_name>" + " \n" + response
 
-    print(final_response)
-    print()
-    print(f"final response type: {type(final_response)}")
-
-    if standing == False:
-        final_response = final_response.replace('standing: 1', 'standing: None')
-    elif hands == False:
-        final_response  = final_response.replace('hands.free: 1', 'hands.free: None')
-    elif location == False:
-        final_response  = final_response.replace('indoors: 1', 'indoors: None')
-    elif screen == False:
-        final_response  = final_response.replace('screen.interaction_yes: 1', 'screen.interaction_yes: None')
-
-    print(final_response)
+    conditions = {
+    'standing': (standing, 'standing: 1', 'standing: None'),
+    'hands': (hands, 'hands.free: 1', 'hands.free: None'),
+    'location': (location, 'indoors: 1', 'indoors: None'),
+    'screen': (screen, 'screen.interaction_yes: 1', 'screen.interaction_yes: None')
+    }
+    
+    for key, (condition, to_replace, replacement) in conditions.items():
+        if not condition:
+            final_response = final_response.replace(to_replace, replacement)
+        
+    # if standing == False:
+    #     final_response = final_response.replace('standing: 1', 'standing: None')
+    # elif hands == False:
+    #     final_response  = final_response.replace('hands.free: 1', 'hands.free: None')
+    # elif location == False:
+    #     final_response  = final_response.replace('indoors: 1', 'indoors: None')
+    # elif screen == False:
+    #     final_response  = final_response.replace('screen.interaction_yes: 1', 'screen.interaction_yes: None')
+
     
     return final_response