Update app.py

app.py

@@ -65,16 +65,17 @@ def load_captioning(uploaded_images, concept_sentence):
         # Update for image component - display image if available, otherwise hide
         image_value = uploaded_images[i - 1] if visible else None
         updates.append(gr.update(value=image_value, visible=visible))
-
-        base_name = image_value.rsplit('.', 1)[0]
         
-        corresponding_txt = base_name + '.txt'
         corresponding_caption = False
-        if corresponding_txt in txt_files:
-            with open(corresponding_txt, 'r') as file:
-                corresponding_caption = file.read()
+        if(image_value):
+            base_name = image_value.rsplit('.', 1)[0]
+            
+            corresponding_txt = base_name + '.txt'
+            if corresponding_txt in txt_files:
+                with open(corresponding_txt, 'r') as file:
+                    corresponding_caption = file.read()
         # Update value of captioning area
-        text_value = corresponding_caption if corresponding_caption else "[trigger]" if visible and concept_sentence else None
+        text_value = corresponding_caption if visible and corresponding_caption else "[trigger]" if visible and concept_sentence else None
         updates.append(gr.update(value=text_value, visible=visible))
 
     # Update for the sample caption area