Update app.py

app.py

@@ -4,37 +4,34 @@ import gradio as gr
 from PIL import Image
 from io import BytesIO
 
-# Function to generate image from Hugging Face API
+# Function to generate image from the Hugging Face API
 def generate_image(prompt):
     API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
-    API_TOKEN = os.getenv("HF_READ_TOKEN")  # Make sure the token is in your environment
+    API_TOKEN = os.getenv("HF_READ_TOKEN")  # Ensure the token is set in your environment
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
     payload = {
-        "inputs": prompt,
+        "inputs": prompt
     }
 
     # Call the Hugging Face API to generate the image
     response = requests.post(API_URL, headers=headers, json=payload)
 
-    # Check if the response was successful
+    # Check if the request was successful
     if response.status_code != 200:
         return f"Error: {response.status_code}, {response.text}"
 
-    # Ensure the response contains an image by loading it into PIL
-    try:
-        image = Image.open(BytesIO(response.content))
-    except Exception as e:
-        return f"Error processing image: {str(e)}"
+    # Convert the response content into a PIL image
+    image = Image.open(BytesIO(response.content))
 
-    return image  # Return the PIL image object
+    return image  # Return the image to Gradio
 
 # Define the chatbot function to return the generated image
 def chatbot(prompt):
     image = generate_image(prompt)
     return image
 
-# Create the Gradio interface
+# Create the Gradio interface with the same UI/UX
 interface = gr.Interface(
     fn=chatbot,
     inputs="text",