update app.py

app.py

@@ -326,9 +326,10 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
     if (task_type == 'inpainting' or task_type == 'remove') and mask_source_radio == mask_source_draw:
         pass
     else:
-        assert text_prompt, 'text_prompt is not found!'
+        assert text_prompt, f'text_prompt for {task_type} is not found!'
 
-    logger.info(f'run_grounded_sam_1_')
+    file_temp = int(time.time())
+    logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_[{text_prompt}]_1_')
 
     # make dir
     os.makedirs(output_dir, exist_ok=True)
@@ -337,9 +338,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
     input_mask = np.array(input_mask_pil.convert("L"))  
      
     image_pil, image = load_image(input_image['image'].convert("RGB"))
-
-    file_temp = int(time.time())
-
+    
     # visualize raw image
     # image_pil.save(os.path.join(output_dir, f"raw_image_{file_temp}.jpg"))
 
@@ -376,7 +375,7 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
         os.remove(image_path)
         output_images.append(detection_image_result)
 
-    logger.info(f'run_grounded_sam_2_')
+    logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_2_')
     if task_type == 'segment' or ((task_type == 'inpainting' or task_type == 'remove') and mask_source_radio == mask_source_segment):
         image = np.array(input_image['image'])
         sam_predictor.set_image(image)
@@ -412,15 +411,15 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
         os.remove(image_path)
         output_images.append(segment_image_result)        
 
-    logger.info(f'run_grounded_sam_3_')
+    logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_3_')
     if task_type == 'detection' or task_type == 'segment':
-        logger.info(f'run_grounded_sam_9_{task_type}_')
+        logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_9_')
         return output_images
     elif task_type == 'inpainting' or task_type == 'remove':
         if inpaint_prompt.strip() == '' and mask_source_radio == mask_source_segment:
             task_type = 'remove'
 
-        logger.info(f'run_grounded_sam_4_{task_type}_')   
+        logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_4_')  
         if mask_source_radio == mask_source_draw:
             mask_pil = input_mask_pil
             mask = input_mask          
@@ -487,12 +486,12 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
         image_inpainting.save(image_path)
         image_result = cv2.cvtColor(cv2.imread(image_path), cv2.COLOR_BGR2RGB)
         os.remove(image_path)
-        logger.info(f'run_grounded_sam_9_{task_type}_')
+        logger.info(f'run_grounded_sam_[{file_temp}]_{task_type}_9_')
         output_images.append(image_result)
         return output_images        
     else:
         logger.info(f"task_type:{task_type} error!")
-    logger.info(f'run_grounded_sam_9_9_')
+    logger.info(f'run_grounded_sam_[{file_temp}]_9_9_')
     return output_images
 
 def change_radio_display(task_type, mask_source_radio):
@@ -525,15 +524,15 @@ if __name__ == "__main__":
                 mask_source_radio = gr.Radio([mask_source_draw, mask_source_segment], 
                                     value=mask_source_segment, label="Mask from",
                                     interactive=True, visible=False)                                                
-                text_prompt = gr.Textbox(label="Detection Prompt", placeholder="Cannot be empty")                                                
+                text_prompt = gr.Textbox(label="Detection Prompt[To detect multiple objects, seperating each name with '.' , Like this: cat . dog . chair ]", placeholder="Cannot be empty")                                                
                 inpaint_prompt = gr.Textbox(label="Inpaint Prompt (if this is empty, then remove)", visible=False)
                 run_button = gr.Button(label="Run")
                 with gr.Accordion("Advanced options", open=False):
                     box_threshold = gr.Slider(
-                        label="Box Threshold", minimum=0.0, maximum=1.0, value=0.3, step=0.001
+                        label="Box Threshold", minimum=0.0, maximum=1.0, value=0.6, step=0.001
                     )
                     text_threshold = gr.Slider(
-                        label="Text Threshold", minimum=0.0, maximum=1.0, value=0.25, step=0.001
+                        label="Text Threshold", minimum=0.0, maximum=1.0, value=0.5, step=0.001
                     )
                     iou_threshold = gr.Slider(
                         label="IOU Threshold", minimum=0.0, maximum=1.0, value=0.5, step=0.001