Update app.py

app.py

@@ -12,26 +12,18 @@ def process_video_and_questions(video, standing, hands, location, screen):
     # Extract the video name (filename)
     video_name = os.path.basename(video)
 
-    standing_flag = None
-    hands_flag = None
-    location_flag = None
-    screen_flag = None
     
     # Construct the query with the video name included
     query = f"Answer the questions from the video\n"
     additional_info = []
     if standing:
         additional_info.append("Is the subject in the video standing or sitting?\n")
-        standing_flag = True
     if hands:
         additional_info.append("Is the subject holding any object in their hands?\n")
-        hands_flag = True
     if location:
         additional_info.append("Is the subject present indoors?\n")
-        location_flag = True
     if screen:
         additional_info.append("Is the subject interacting with a screen in the background by facing the screen?\n")
-        screen_flag = True
     
     end_query = """Provide the results in <annotation> tags, where 0 indicates False, 1 indicates True, and None indicates that no information is present. Follow the below examples\n:
         <annotation>indoors: 0</annotation>
@@ -51,8 +43,14 @@ def process_video_and_questions(video, standing, hands, location, screen):
     print()
     print(f"final response type: {type(final_response)}")
 
-    if standing_flag == None:
+    if standing == False:
         final_response.replace('standing: 1', 'standing: None')
+    elif hands == False:
+        final_response.replace('hands.free: 1', 'hands.free: None')
+    elif location == False:
+        final_response.replace('indoors: 1', 'indoors: None')
+    elif screen == False:
+        final_response.replace('screen.interaction_yes: 1', 'screen.interaction_yes: None')
 
     print(final_response)