Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import requests
 import io
@@ -27,13 +28,18 @@ def set_system_prompt():
 # Function to enhance the prompt with Qwen model
 def enhance_prompt_with_qwen(prompt):
     client = Client("Qwen/Qwen2.5-72B-Instruct")
-    result,_,__ = client.predict(
+    result = client.predict(
         query=prompt,
         history=[],
         system="You are Qwen, an image generation prompt enhancer",
         api_name="/model_chat"
     )
-    return str(result) # Assuming the enhanced prompt is under 'output'
+    
+    # Extract the relevant part of the tuple, index [0], which contains the enhanced prompt.
+    enhanced_prompt = result[0]  # This is the string we need for the image generation prompt.
+    
+    print(f"Enhanced prompt: {enhanced_prompt}")
+    return enhanced_prompt
 
 # Image generation query function
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, huggingface_api_key=None, use_dev=False):