Update app.py

app.py

@@ -1,31 +1,30 @@
-import gradio as gr
-import requests
 import os
-import json
+import requests
+import gradio as gr
 
-# Define the function that queries the Hugging Face API
-def generate_image_from_model(prompt):
+# Define the function to query Hugging Face API for image generation
+def generate_image(prompt):
     API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
-    API_TOKEN = os.getenv("HF_READ_TOKEN")
+    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)
 
-    # Handle errors
+    # Check if the request was successful
     if response.status_code != 200:
         return f"Error: {response.status_code}, {response.text}"
 
-    # Assuming the response is an image, save it temporarily and return the path
-    image_data = response.content
+    # Save the generated image
     image_path = "generated_image.png"
-    with open(image_path, "wb") as img_file:
-        img_file.write(image_data)
+    with open(image_path, "wb") as f:
+        f.write(response.content)
 
-    return image_path  # Return the path to the image
+    return image_path
 
 # Define the chatbot function to return the generated image
 def chatbot(prompt):