Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 from io import BytesIO
 from PIL import Image
 import os
-# HARDCODED_URL = ""  # Replace with your desired URL
 
 TOKEN = os.getenv("TOKEN1")
 API_URL = os.getenv("API_URL")
@@ -14,13 +13,23 @@ model_id = os.getenv("MODEL_ID")
 
 def get_image_from_url(url):
     """
-    Fetches and returns an image from a given URL.
+    Fetches and returns an image from a given URL, converting to PNG if needed.
     """
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
         image = Image.open(BytesIO(response.content))
+        
+        # Convert to PNG if the image is WebP
+        if image.format == "WEBP":
+            image = image.convert("RGBA")  # Convert to RGBA to handle transparency
+            output_buffer = BytesIO()
+            image.save(output_buffer, format="PNG")
+            output_buffer.seek(0)  # Reset buffer position for reading
+            image = Image.open(output_buffer) # Re-open image from buffer
+        
         return image
+
     except requests.exceptions.RequestException as e:
         return f"Error fetching image: {e}"
     except Exception as e:
@@ -33,7 +42,7 @@ def generate_image(prompt, aspect_ratio, realism):
     global no_of_accounts
     global model_id
     payload = {
-        "id": model_id,  # Replace with appropriate ID if necessary
+        "id": model_id,  
         "inputs": [prompt, aspect_ratio, str(realism).lower()],
     }
     headers = {"Authorization": f"Bearer {TOKEN}"}
@@ -59,7 +68,7 @@ def generate_image(prompt, aspect_ratio, realism):
     except Exception as e:
         return f"Error"
 
-# Define the Gradio interface
+
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
@@ -77,4 +86,4 @@ interface = gr.Interface(
 )
 
 # Launch the interface
-interface.launch()
+interface.launch()