Update app.py

app.py

@@ -69,18 +69,21 @@ def process_image_and_text(subject_image, adapter_dict, prompt, _unused1, _unuse
     if BASETEN_API_KEY:
         headers["Authorization"] = f"Api-Key {BASETEN_API_KEY}"
 
-    resp = requests.post(BASETEN_MODEL_URL, headers=headers, json=payload, timeout=120)
+    resp = requests.post(BASETEN_MODEL_URL, headers=headers, json=payload, timeout=180)
     resp.raise_for_status()
-
-    if resp.headers.get("content-type", "").startswith("image/"):
-        raw_img = Image.open(BytesIO(resp.content))
+    data = resp.json()
+    print (data)
+    
+    # If Baseten returns [cond1.condition, raw_img] format
+    if isinstance(data, list) and len(data) == 2:
+        # If the second item is base64-encoded image string
+        try:
+            raw_img = Image.open(BytesIO(base64.b64decode(data[1])))
+        except Exception:
+            raise gr.Error("Failed to decode image from Baseten response.")
     else:
-        url = resp.json().get("image_url")
-        if not url:
-            raise gr.Error("Baseten response missing image data.")
-        raw_img = Image.open(BytesIO(requests.get(url, timeout=120).content))
+        raise gr.Error("Unexpected response format from Baseten.")
 
-    return [[raw_img]], raw_img
 
 # ────────── Header HTML ──────────
 header_html = """