Update app.py

app.py

@@ -68,7 +68,7 @@ def load_captioning(uploaded_images, concept_sentence):
         
         corresponding_caption = False
         if(image_value):
-            base_name = os.path.splitext(os.path.basename(image))[0]
+            base_name = os.path.splitext(os.path.basename(image_value))[0]
             if base_name in txt_file_dict:
                 with open(txt_file_dict[base_name], 'r') as file:
                     corresponding_caption = file.read()
@@ -85,6 +85,9 @@ def load_captioning(uploaded_images, concept_sentence):
     updates.append(gr.update(placeholder=f"A {concept_sentence} in a mall"))
     return updates
 
+def hide_captioning():
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) 
+
 def create_dataset(*inputs):
     print("Creating dataset")
     images = inputs[0]
@@ -464,6 +467,10 @@ with gr.Blocks(theme=theme, css=css) as demo:
         inputs=[steps],
         outputs=[cost_preview, cost_preview_info, payment_update, start]
     )
+    images.clear(
+        hide_captioning,
+        outputs=[captioning_area, cost_preview, start]
+    )
     gr.on(
         triggers=[steps.change, payment_update.click],
         fn=update_pricing,