Update app.py

app.py

@@ -209,7 +209,7 @@ def generate_textual_description(box_info):
     kept_zero_boxes.sort(key = lambda x: x[0])
     kept_one_boxes.sort(key = lambda x: x[0])
     textual_description = "" #final output
-    textual_description += "The key elements are interpreted via visual translator. Their coordinates are represented as outlined boxes (top-left, bottom-right). "
+    textual_description += "The key elements are interpreted via visual translator. Their coordinates are represented as outlined boxes (top-left, bottom-right)."
     #print(f"The key elements are interpreted via visual translator. Their coordinates are represented as outlined boxes (top-left, bottom-right)")
     if len(kept_zero_boxes) >= 1:
         left_most_zero_cor = kept_zero_boxes[0]
@@ -246,7 +246,8 @@ def drawWithAllBox_info(pil_image, box_info):
 
 def process_image(image):
     if image is None:
-        return None, "", "", None
+        # Ensure we always return 3 values to prevent errors.
+        return None, "", ""
     pil_image = image.copy() if hasattr(image, 'copy') else Image.fromarray(image)
     roboflow_ret = RoboFlowGetOutlineBoxesPIL(pil_image)
     all_box_info = parse_roboflow_result(roboflow_ret, classes)
@@ -279,15 +280,13 @@ with gr.Blocks() as demo:
         text_out = gr.Textbox(label="Textual Description", lines=8)
         json_state = gr.State("")
         download_btn = gr.DownloadButton(
-            label="Download Box Info as JSON"
+            label="Download Box Info as JSON",
+            interactive=False  # Start as disabled.
         )
 
     # --- Backend Functions ---
-    def _process(image):
-        boxed_img, textual, json_str = process_image(image)
-        return boxed_img, textual, json_str
-
-    def create_and_download_json(json_str):
+    def create_json_file(json_str):
+        """Creates a temp file with JSON content and returns its path."""
         if not json_str:
             return None
         with tempfile.NamedTemporaryFile(
@@ -297,6 +296,17 @@ with gr.Blocks() as demo:
             f.write(json_str)
             return f.name
 
+    def _process_and_prepare_download(image):
+        """Processes the image, creates the JSON file, and updates the UI."""
+        boxed_img, textual, json_str = process_image(image)
+        filepath = create_json_file(json_str)
+        
+        # Create an update object for the download button.
+        # This sets its value to the file path and makes it clickable.
+        download_update = gr.DownloadButton.update(value=filepath, interactive=True)
+        
+        return boxed_img, textual, json_str, download_update
+
     def check_token(token):
         # Securely check if the token is correct
         if ACCESS_TOKEN and token == ACCESS_TOKEN:
@@ -312,15 +322,13 @@ with gr.Blocks() as demo:
     )
 
     run_btn.click(
-        _process,
+        _process_and_prepare_download,
         inputs=img_input,
-        outputs=[img_out, text_out, json_state]
+        # The output now includes the download button itself.
+        outputs=[img_out, text_out, json_state, download_btn]
     )
 
-    download_btn.click(
-        create_and_download_json,
-        inputs=json_state,
-        outputs=download_btn
-    )
+    # The download_btn no longer needs its own click event.
+    # Clicking it after it has a 'value' will trigger the download.
 
 demo.launch()