Update app.py

app.py

@@ -708,7 +708,7 @@ def build_ui_with_custom_ui():
             if img:
                 captions = captions + [captioner(img)[0]["generated_text"]]
             ready = len(images) >= 1 and len(captions) >= 1
-            return img, images, captions, gr.update(interactive=ready), gr.update(interactive=ready)
+            return img, images, captions #,gr.update(interactive=ready), gr.update(interactive=ready)
 
         def generate_ds(_, enhancer, images, captions):
             if not captions:
@@ -718,7 +718,7 @@ def build_ui_with_custom_ui():
             if img:
                 captions = captions + [captioner(img)[0]["generated_text"]]
             ready = len(images) >= 1 and len(captions) >= 1
-            return img, images, captions, gr.update(interactive=ready), gr.update(interactive=ready)
+            return img, images, captions  #, gr.update(interactive=ready), gr.update(interactive=ready)
 
         
   
@@ -783,15 +783,25 @@ def build_ui_with_custom_ui():
 
 
         # ---------------- Enable Metrics/NLP only when ready ----------------
-
+        """
         def enable_metrics_nlp(images, captions):
           ready = len(images) >= 3 and len(captions) >= 3
           return (
               gr.update(interactive=ready),
               gr.update(interactive=ready)
-          )
-     
-        #images_state.change(enable_metrics_nlp, [images_state, captions_state], [metrics_btn, nlp_btn])
+          )"""
+
+        def enable_metrics_nlp(images, captions):
+            ready = (
+                len(images) == 3 and
+                len(captions) == 3 and
+                all(c and c != "Caption failed." for c in captions)
+            )
+            return gr.update(interactive=ready), gr.update(interactive=ready)
+
+
+                
+        images_state.change(enable_metrics_nlp, [images_state, captions_state], [metrics_btn, nlp_btn])
 
         # ---------------- Step 5: VQA ----------------
         gr.Markdown("### Visual Question Answering (VQA)", elem_classes="heading-orange")