Update app.py

app.py

@@ -44,7 +44,7 @@ Don't write anything else except the character description in json format and do
 world_description_prompt = "Generate a unique and random world description (Don't Write anything else except the world description)."
 
 def get_random_world_description():
-    result = llm_client.chat.completions.create(
+    response = llm_client.chat.completions.create(
         model="Qwen/Qwen3-235B-A22B",
         messages=[
             {
@@ -53,7 +53,10 @@ def get_random_world_description():
             }
         ],
     )
-    return result.choices[0].message
+    result = response.choices[0].message.content
+    if "</think>" in result:
+        result = result[result.index("</think>")+1:]
+    return result.choices[0].message.content
 
 def get_random_persona_description():
     return ds.shuffle().select([100])[0]["persona"]
@@ -84,12 +87,15 @@ def generate_character(world_description, persona_description, progress=gr.Progr
             }
         ],
     )
-    output = json.loads(result.choices[0].message)
+    result = response.choices[0].message.content
+    if "</think>" in result:
+        result = result[result.index("</think>")+1:]
+    output = json.loads(result)
     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 character's appearance is generated using [FLUX-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev) and the character description is generated using [Qwen3-235B-A22B](https://huggingface.co/Qwen/Qwen3-235B-A22B).
 - 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.