Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import importlib
 from PIL import Image
 import json
+import tempfile
 import spaces
 
 # === Model Mapping ===
@@ -142,25 +143,28 @@ def format_pretty_view(output):
 def process_single_image(model_name, image_file):
     runner = load_model_runner(model_name)
     image = Image.open(image_file.name).convert("RGB")
-    result = runner(image)
+    base_name = os.path.splitext(os.path.basename(image_file.name))[0]
 
+    result = runner(image)
     parsed_json = result.get("json")
     raw_text = result.get("raw", "")
 
     if parsed_json:
-        json_output = format_result_json(parsed_json)
+        json_output = json.dumps(parsed_json, indent=2)
         pretty_output = format_pretty_view(parsed_json)
 
-        # Save JSON to a temporary file for download
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".json", mode="w", encoding="utf-8") as tmp:
-            json.dump(parsed_json, tmp, indent=2)
-            json_path = tmp.name
+        tmp_path = os.path.join(tempfile.gettempdir(), f"{base_name}_output.json")
+        with open(tmp_path, "w", encoding="utf-8") as f:
+            json.dump(parsed_json, f, indent=2)
     else:
         json_output = "(No valid JSON extracted)"
         pretty_output = "(No structured content extracted)\n\n⚠️ Raw Model Output:\n" + raw_text
-        json_path = None
 
-    return image, json_output, pretty_output, json_path
+        tmp_path = os.path.join(tempfile.gettempdir(), f"{base_name}_output.txt")
+        with open(tmp_path, "w", encoding="utf-8") as f:
+            f.write(raw_text)
+
+    return image, json_output, pretty_output, tmp_path
 
 # === Gradio UI
 iface = gr.Interface(