Update app.py

app.py

@@ -24,6 +24,22 @@ def get_headers():
         "X-HF-Bill-To": "huggingface"
     }
 
+def query_api(payload):
+    """Send request to the API and return response"""
+    headers = get_headers()
+    
+    # Convert image to base64 if it's bytes
+    if "image_bytes" in payload:
+        payload["inputs"] = base64.b64encode(payload["image_bytes"]).decode("utf-8")
+        del payload["image_bytes"]
+    
+    response = requests.post(API_URL, headers=headers, json=payload)
+    
+    if response.status_code != 200:
+        raise gr.Error(f"API request failed with status {response.status_code}: {response.text}")
+    
+    return response.content
+
 def upload_image_to_fal(image_bytes):
     """Upload image to fal.ai and return the URL"""
     # For now, we'll use base64 data URI as mentioned in the docs