Update app.py

app.py

@@ -39,9 +39,9 @@ MAX_IMAGES = 150
 
 
 def load_captioning(uploaded_images, concept_sentence):
-    
     uploaded_images = [file for file in uploaded_images if not file.endswith('.txt')]
     txt_files = [file for file in uploaded_images if file.endswith('.txt')]
+    txt_files_dict = {os.path.splitext(os.path.basename(txt_file))[0]: txt_file for txt_file in txt_files}
     
     gr.Info("Images uploaded!")
     updates = []
@@ -67,15 +67,11 @@ def load_captioning(uploaded_images, concept_sentence):
         updates.append(gr.update(value=image_value, visible=visible))
         
         corresponding_caption = False
-        print(txt_files)
         if(image_value):
-            print(image_value)
-            base_name = image_value.rsplit('.', 1)[0]
-            corresponding_txt = base_name + '.txt'
-            print(corresponding_txt)
-            if corresponding_txt in txt_files:
-                with open(corresponding_txt, 'r') as file:
-                    corresponding_caption = file.read()
+            base_name = os.path.splitext(os.path.basename(image))[0]
+            if base_name in txt_file_dict:
+            with open(txt_file_dict[base_name], 'r') as file:
+                corresponding_caption = file.read()
         # Update value of captioning area
         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))