Update app.py

app.py

@@ -1,17 +1,22 @@
+## app.py:
 import torch
 import gradio as gr
+from diffusers import StableDiffusionPipeline
 import requests
 from io import BytesIO
-from PIL import Image
 import os
+from PIL import Image
+
+
 
 def translate_text(text, target_language='en'):
     API_URL = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-ar-en"
     headers = {"Authorization": f"Bearer {os.getenv('API_TOKEN')}"}
-    response = requests.post(API_URL, headers=headers, json={"inputs": text})
+    response = requests.post(API_URL, headers=headers, json=text)
 
     if response.status_code == 200:
         return response.json()[0]['translation_text']
+
     else:
         print("Failed to translate text:", response.text)
         return text  # Return the original text if translation fails
@@ -25,13 +30,15 @@ def query(payload, API_URL, headers):
 def generate_image(prompt, model_choice, translate=False):
     if translate:
         prompt = translate_text(prompt, target_language='en')  # Assuming you want to translate to English
-
     model_urls = {
         "Stable Diffusion v1.5": "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
         "dalle-3-xl-v2": "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl-v2",
         "midjourney-v6": "https://api-inference.huggingface.co/models/Kvikontent/midjourney-v6",
         "openjourney-v4": "https://api-inference.huggingface.co/models/prompthero/openjourney-v4",
         "LCM_Dreamshaper_v7": "https://api-inference.huggingface.co/models/SimianLuo/LCM_Dreamshaper_v7",
+
+
+
     }
     API_URL = model_urls[model_choice]
 
@@ -43,6 +50,10 @@ def generate_image(prompt, model_choice, translate=False):
         image = Image.open(BytesIO(data))
         # Resize the image
         image = image.resize((400, 400))
+        # Convert the image object back to bytes for Gradio output
+        buf = BytesIO()
+        image.save(buf, format='PNG')
+        buf.seek(0)
         return image
 
     except Exception as e:
@@ -65,11 +76,11 @@ description = "اكتب وصف للصورة التي تود من النظام ا
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(lines=2, placeholder="Enter the description of the image here..."),
-        gr.Dropdown(choices=["Stable Diffusion v1.5","dalle-3-xl-v2","midjourney-v6","openjourney-v4","LCM_Dreamshaper_v7"], label="Choose Model", value='Stable Diffusion v1.5'),
-        gr.Checkbox(label="Translate The Text Before Generating Image", value=False)
+        gr.components.Textbox(lines=2, placeholder="Enter the description of the image here..."),
+        gr.components.Dropdown(choices=["Stable Diffusion v1.5","dalle-3-xl-v2","midjourney-v6","openjourney-v4","LCM_Dreamshaper_v7"], label="Choose Model", value='Stable Diffusion v1.5'),
+
     ],
-    outputs=gr.Image(),
+    outputs=gr.components.Image(),
     title=title,
     description=description,
     theme="default",