update output path logic

app.py

@@ -15,19 +15,24 @@ import spaces
 hf_token = os.getenv("HF_TOKEN")
 login(token=hf_token, add_to_git_credential=True)
 
-def detect_brain_tumor(image_path: str, output_path: str, debug: bool = False) -> None:
+def detect_brain_tumor(image, debug: bool = False) -> str:
     """
     Detects a brain tumor in the given image and saves the image with bounding boxes.
 
     Parameters:
-        image_path (str): The path to the input image.
-        output_path (str): The path to save the output image with bounding boxes.
+        image: The input image (as a PIL Image or numpy array).
         debug (bool): Flag to enable logging for debugging purposes.
+
+    Returns:
+        str: Path to the saved output image.
     """
-    # Step 1: Load the image
-    image = load_image(image_path)
+    # Generate a unique output filename
+    output_path = f"./output/tumor_detection_{int(time.time())}.jpg"
+    
+    # Step 1: Load the image (not needed if image is already a PIL Image or numpy array)
+    # image = load_image(image_path)
     if debug:
-        print(f"Image loaded from {image_path}")
+        print(f"Image loaded")
 
     # Step 2: Detect brain tumor using owl_v2
     prompt = "detect brain tumor"
@@ -44,6 +49,9 @@ def detect_brain_tumor(image_path: str, output_path: str, debug: bool = False) -
     save_image(image_with_bboxes, output_path)
     if debug:
         print(f"Image saved to {output_path}")
+    
+    return output_path
+
 
 # Example usage (uncomment to run):
 # detect_brain_tumor("/content/drive/MyDrive/kaggle/datasets/brain-tumor-image-dataset-semantic-segmentation_old/train_categories/1385_jpg.rf.3c67cb92e2922dba0e6dba86f69df40b.jpg", "/content/drive/MyDrive/kaggle/datasets/brain-tumor-image-dataset-semantic-segmentation_old/output/1385_jpg.rf.3c67cb92e2922dba0e6dba86f69df40b.jpg", debug=True)
@@ -65,9 +73,8 @@ with gr.Blocks(css="style.css") as demo:
         chat_btn = gr.Button()
 
         chat_inputs = [
-            image,
-            "./output/tmp.jpg",
-            ]
+            image
+        ]
         chat_outputs = [
             text_output
         ]