Update app.py

app.py

@@ -15,15 +15,14 @@ HEADERS = {
 }
 
 def generate_or_edit(prompt, input_image=None):
-    # Flux API के लिए मानक payload स्ट्रक्चर
     payload = {
         "prompt": prompt,
-        "aspect_ratio": "1:1",
-        "seed": 42,
+        "width": 1024,
+        "height": 1024,
+        "seed": 0,
         "steps": 4
     }
 
-    # अगर इमेज एडिटिंग के लिए है
     if input_image is not None:
         buffered = io.BytesIO()
         input_image.save(buffered, format="PNG")
@@ -35,19 +34,12 @@ def generate_or_edit(prompt, input_image=None):
         response.raise_for_status()
         result = response.json()
         
-        # NVIDIA API अक्सर 'result' या 'image' की में इमेज भेजता है
-        if "image" in result:
-            image_data = base64.b64decode(result["image"])
-        elif "output" in result and "image" in result["output"]:
-            image_data = base64.b64decode(result["output"]["image"])
-        else:
-            return None
+        # अब हम सीधे उस स्ट्रक्चर का उपयोग कर रहे हैं जो आपने दिखाया है
+        image_b64 = result["artifacts"][0]["base64"]
+        image_data = base64.b64decode(image_b64)
             
         return Image.open(io.BytesIO(image_data))
         
-    except requests.exceptions.HTTPError as e:
-        print(f"HTTP Error: {e.response.text}") # यहाँ असली एरर दिखेगा
-        return None
     except Exception as e:
         print(f"Error: {e}")
         return None
@@ -62,4 +54,4 @@ with gr.Blocks() as demo:
 
     submit_btn.click(fn=generate_or_edit, inputs=[prompt_input, image_input], outputs=result_output)
 
-demo.launch()
+demo.launch(server_name="0.0.0.0", server_port=7860)