backup

app.py

@@ -305,18 +305,31 @@ with gr.Blocks() as demo:
             zip(response_data[::3], response_data[1::3], response_data[2::3])
         )
 
-        # Filter out empty responses (only checking image and model name, notes are optional)
+        # Check for responses with images but no model names
+        incomplete_responses = [
+            i + 1
+            for i, (img, txt, _) in enumerate(response_triplets)
+            if img is not None and not txt.strip()
+        ]
+
+        if incomplete_responses:
+            gr.Warning(
+                f"Please provide model names for responses: {', '.join(map(str, incomplete_responses))}!"
+            )
+            return
+
+        # Filter out empty responses (where both image and model name are empty)
         valid_responses = [
             (img, txt, notes)
             for img, txt, notes in response_triplets
-            if img is not None and txt
+            if img is not None and txt.strip()
         ]
 
         if not valid_responses:
-            gr.Warning("Please provide at least one response (image + text)!")
+            gr.Warning("Please provide at least one response (image + model name)!")
             return
 
-        # Modify generate_json_files call to handle notes
+        # Generate JSON files with the valid responses
         generate_json_files(source_img, valid_responses, post_id)
         gr.Info("Responses saved successfully! 🎉")