Update app.py

app.py

@@ -1,46 +1,39 @@
 import os
 import requests
-from flask import Flask, request, jsonify, render_template
+import gradio as gr
 
-app = Flask(__name__)
-
-# Hugging Face API Configuration
+# Hugging Face API
 HF_API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
-HF_API_KEY = os.environ.get("HF_TOKEN")  # Set your Hugging Face API key
+HF_API_KEY = os.getenv("HF_API_KEY")
 
-# Chevereto Configuration
-CHEVERETO_API_URL = os.environ.get("API_URL")
-CHEVERETO_API_KEY = os.environ.get("API_KEY")
-CHEVERETO_ALBUM_ID = os.environ.get("ALBUM_ID")
+# Chevereto API
+CHEVERETO_API_URL = os.getenv("API_URL")
+CHEVERETO_API_KEY = os.getenv("API_KEY")
+CHEVERETO_ALBUM_ID = os.getenv("ALBUM_ID")
 
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
-def generate_image_and_upload(prompt: str) -> str:
-    """Generates an image using Hugging Face API and uploads it to Chevereto."""
-    
+def generate_image(prompt: str):
+    """Generates an image using Hugging Face's API"""
     response = requests.post(HF_API_URL, headers=HEADERS, json={"inputs": prompt})
-
+    
     if response.status_code == 200:
         image_data = response.content
-        image_path = "static/generated_image.png"
-        
+        image_path = "/tmp/generated_image.png"
+
         with open(image_path, "wb") as f:
             f.write(image_data)
 
-        return upload_to_chevereto(image_path)  # Upload and return URL
+        return image_path
     else:
         return f"Error generating image: {response.text}"
 
-def upload_to_chevereto(img_path: str) -> str:
-    """Uploads an image to Chevereto and returns the public image URL."""
-    
-    with open(img_path, "rb") as image_file:
-        files = {"source": image_file}
-        data = {
-            "key": CHEVERETO_API_KEY,
-            "format": "json",
-            "album": CHEVERETO_ALBUM_ID
-        }
+def upload_to_chevereto(image_path: str):
+    """Uploads the generated image to Chevereto and returns the URL"""
+    with open(image_path, "rb") as img_file:
+        files = {"source": img_file}
+        data = {"key": CHEVERETO_API_KEY, "format": "json", "album": CHEVERETO_ALBUM_ID}
+        
         response = requests.post(CHEVERETO_API_URL, files=files, data=data)
     
     if response.status_code == 200:
@@ -48,22 +41,27 @@ def upload_to_chevereto(img_path: str) -> str:
     else:
         return f"Error uploading image: {response.text}"
 
-@app.route("/", methods=["GET"])
-def index():
-    """Renders the Star Trek-themed UI."""
-    return render_template("index.html")
+def generate_and_upload(prompt: str):
+    """Combines image generation + upload and returns Chevereto URL"""
+    img_path = generate_image(prompt)
+    if "Error" in img_path:
+        return img_path
+    return upload_to_chevereto(img_path)
 
-@app.route("/generate", methods=["POST"])
-def generate():
-    """API Endpoint: Generate an image from a text prompt and upload it."""
-    data = request.get_json()
-    prompt = data.get("prompt", "")
-
-    if not prompt:
-        return jsonify({"error": "Missing prompt"}), 400
+# Gradio UI
+with gr.Blocks() as app:
+    gr.Markdown("<h1 style='text-align: center; color: red;'>🚀 Star Trek AI Image Generator</h1>")
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Enter a prompt:", placeholder="A futuristic spaceship...")
+        generate_button = gr.Button("Generate & Upload")
+    
+    image_output = gr.Image(label="Generated Image")
+    url_output = gr.Textbox(label="Image URL", interactive=False)
 
-    image_url = generate_image_and_upload(prompt)
-    return jsonify({"image_url": image_url})
+    generate_button.click(
+        fn=generate_and_upload,
+        inputs=prompt_input,
+        outputs=[image_output, url_output]
+    )
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7680)
+app.launch(server_port=7860)