chore: Refactor character generation and image inference in app.py

app.py

@@ -55,22 +55,10 @@ def get_random_world_description():
 def get_random_persona_description():
     return ds.shuffle().select([100])[0]["persona"]
 
-@spaces.GPU()
-def generate_character(world_description, persona_description, progress=gr.Progress(track_tqdm=True)):
-    result = llm_client.predict(
-        query=prompt_template.format(
-            persona_description=persona_description, world_description=world_description
-        ),
-        history=[],
-        system="You are Qwen, created by Alibaba Cloud. You are a helpful assistant.",
-        api_name="/model_chat",
-    )
-    output = json.loads(result[1][0][-1])
-    print("Character generated")
-    print(json.dumps(output, indent=4))
-
+@spaces.GPU(duration=75)
+def infer_flux(character_json):
     for image in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
-        prompt=output["appearance"],
+        prompt=character_json["appearance"],
         guidance_scale=3.5,
         num_inference_steps=28,
         width=1024,
@@ -79,14 +67,26 @@ def generate_character(world_description, persona_description, progress=gr.Progr
         output_type="pil",
         good_vae=good_vae,
     ):
-        yield image, output
-    print("Character and image generated")
+        yield image
 
+def generate_character(world_description, persona_description, progress=gr.Progress(track_tqdm=True)):
+    result = llm_client.predict(
+        query=prompt_template.format(
+            persona_description=persona_description, world_description=world_description
+        ),
+        history=[],
+        system="You are Qwen, created by Alibaba Cloud. You are a helpful assistant.",
+        api_name="/model_chat",
+    )
+    output = json.loads(result[1][0][-1])
+    return output
 
 app_description = """
 - This app generates a character in JSON format based on a persona description and a world description.
 - The character's appearance is generated using [FLUX-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev) and the character description is generated using [Qwen2.5-72B-Instruct](https://huggingface.co/Qwen/Qwen2.5-72B-Instruct).
 - The persona description is randomly selected from the [FinePersonas-Lite](https://huggingface.co/datasets/MohamedRashad/FinePersonas-Lite) dataset.
+
+**Note:** I recommend starting with the world description (you can write one or loop over randomly generated ones) and then try different persona descriptions to generate interesting characters for the world you created.
 """
 
 with gr.Blocks(title="Character Generator") as app:
@@ -103,7 +103,7 @@ with gr.Blocks(title="Character Generator") as app:
                 random_persona_button = gr.Button(value="Get Random Persona Description", variant="secondary", scale=1)
         with gr.Row():
             character_image = gr.Image(label="Character Image")
-            character = gr.JSON(label="Character Description")
+            character_json = gr.JSON(label="Character Description")
 
         examples = gr.Examples(
             [
@@ -115,8 +115,8 @@ with gr.Blocks(title="Character Generator") as app:
         )
 
     submit_button.click(
-        generate_character, [world_description, persona_description], outputs=[character_image, character]
-    )
+        generate_character, [world_description, persona_description], outputs=[character_json]
+    ).then(fn=infer_flux, inputs=[character_json], outputs=[character_image])
     random_world_button.click(
         get_random_world_description, outputs=[world_description]
     )