enhance text generation: progress and error message

app.py

@@ -65,7 +65,7 @@ def get_text_gen_pipeline():
     return _text_gen_pipeline
 
 @spaces.GPU()
-def refine_prompt(prompt):
+def refine_prompt(prompt, progress=gr.Progress(track_tqdm=True)):
     text_gen = get_text_gen_pipeline()
     if text_gen is None:
         return "Text generation model is unavailable."
@@ -73,7 +73,9 @@ def refine_prompt(prompt):
         messages = [
             {"role": "system", "content": "Vous êtes un designer produit avec de solides connaissances dans la génération de texte en image. Vous recevrez une demande de produit sous forme de description succincte, et votre mission sera d'imaginer un nouveau design de produit répondant à ce besoin.\n\nLe livrable (réponse générée) sera exclusivement un texte de prompt pour l'IA de texte to image FLUX.1-schnell.\n\nCe prompt devra inclure une description visuelle de l'objet mentionnant explicitement les aspects indispensables de sa fonction.\nA coté de ça vous devez aussi explicitement mentionner dans ce prompt les caractéristiques esthétiques/photo du rendu image (ex : photoréaliste, haute qualité, focale, grain, etc.), sachant que l'image sera l'image principale de cet objet dans le catalogue produit. Le fond de l'image générée doit être entièrement blanc.\nLe prompt doit être sans narration, peut être long mais ne doit pas dépasser 77 jetons."},            {"role": "user", "content": prompt},
         ]
-        refined_prompt = text_gen(messages)
+        with progress.tqdm(total=1, desc="Generating text") as pbar:
+            refined_prompt = text_gen(messages)
+            pbar.update(1)
         
         # Extract just the assistant's content from the response
         try:
@@ -83,11 +85,11 @@ def refine_prompt(prompt):
             if not assistant_messages:
                 return "Error: No assistant response found"
             assistant_content = assistant_messages[-1]['content']
-            return assistant_content
+            return assistant_content, "Prompt refined successfully!"
         except (KeyError, IndexError):
-            return "Error: Unexpected response format from the model"
+            return "", "Error: Unexpected response format from the model"
     except Exception as e:
-        return f"Error refining prompt: {str(e)}"
+        return "", f"Error refining prompt: {str(e)}"
 
 def validate_dimensions(width, height):
     if width * height > MAX_IMAGE_SIZE * MAX_IMAGE_SIZE:
@@ -254,7 +256,7 @@ def create_interface():
             triggers=[prompt_button.click, prompt.submit],
             fn=refine_prompt,
             inputs=[prompt],
-            outputs=[refined_prompt]
+            outputs=[refined_prompt, error_box]
         )
 
         gr.on(