Switch image generation from Ideogram to FLUX.1-schnell

app.py

@@ -1,7 +1,6 @@
 import os
 import json
 import gradio as gr
-from gradio_client import Client
 from huggingface_hub import InferenceClient
 
 
@@ -16,7 +15,7 @@ def generate_prompt(concept: str) -> str:
         system_instruction = (
             "You are an expert prompt engineer for text-to-image models. "
             "Your task is to take a simple concept and expand it into a detailed, "
-            "vivid, and high-quality image prompt for Ideogram. "
+            "vivid, and high-quality image prompt for FLUX.1-schnell. "
             "Describe the scene, lighting, materials, and aesthetic in detail. "
             "Provide ONLY the final prompt text. Do not include any introductory or concluding text, "
             "do not provide multiple options, and do not wrap the prompt in quotes."
@@ -44,35 +43,29 @@ def generate_prompt(concept: str) -> str:
 
 def generate_image(prompt: str) -> dict:
     """
-    Generates an image from a prompt using the Ideogram 4 model.
+    Generates an image from a prompt using the FLUX.1-schnell model.
     Returns a dictionary structure compatible with Gradio's image viewer.
     """
     if not prompt:
         prompt = "a ginger cat wearing a tiny wizard hat reading a spellbook"
     try:
-        client = Client("ideogram-ai/ideogram4", verbose=False)
-        result = client.predict(
-            prompt=prompt,
-            mode="Default · 20 steps",
-            upsampler="Ideogram (remote)",
-            width=1024,
-            height=1024,
-            seed=0,
-            randomize_seed=True,
-            api_name="/generate"
-        )
-        if isinstance(result, (list, tuple)) and len(result) > 0:
-            filepath = result[0]
-        else:
-            filepath = result
-
+        client = InferenceClient("black-forest-labs/FLUX.1-schnell")
+        image = client.text_to_image(prompt)
+        
+        import tempfile
+        import uuid
+        
+        temp_dir = tempfile.gettempdir()
+        filepath = os.path.join(temp_dir, f"{uuid.uuid4()}.png")
+        image.save(filepath)
+        
         return {
             "path": filepath,
             "url": f"/gradio_api/file={filepath}",
             "is_file": True
         }
     except Exception as e:
-        print(f"Error calling Ideogram 4 space: {e}")
+        print(f"Error calling FLUX.1-schnell model: {e}")
         raise e
 
 

workflow.json

@@ -1,6 +1,6 @@
 {
   "schema_version": "2",
-  "name": "Ideogram 4 Workflow",
+  "name": "FLUX.1-schnell Workflow",
   "runtime": {
     "default": "client"
   },
@@ -90,7 +90,7 @@
     },
     {
       "id": "op_generate_image",
-      "label": "Ideogram 4 Image Generator",
+      "label": "FLUX.1-schnell Image Generator",
       "role": "operator",
       "kind": "fn",
       "fn": "generate_image",