refactor prompt refinement and interface output handling for improved clarity

app.py

@@ -111,6 +111,11 @@ def refine_prompt(prompt, system_prompt=DEFAULT_SYSTEM_PROMPT, progress=gr.Progr
             if not assistant_messages:
                 return "", "Error: No assistant response found"
             assistant_content = assistant_messages[-1]['content']
+            
+            # Remove quotation marks at the beginning and end
+            if assistant_content.startswith('"') and assistant_content.endswith('"'):
+                assistant_content = assistant_content[1:-1]
+                
             return assistant_content, "Prompt refined successfully!"
         except (KeyError, IndexError):
             return "", "Error: Unexpected response format from the model"
@@ -206,10 +211,10 @@ def process_example_pipeline(example_prompt, system_prompt=DEFAULT_SYSTEM_PROMPT
     refined, status = refine_prompt(example_prompt, system_prompt, progress)
     
     if not refined:
-        return example_prompt, "", "Failed to refine prompt: " + status
+        return "", "Failed to refine prompt: " + status
     
     # Return only the refined prompt and status - don't generate image
-    return example_prompt, refined, "Prompt refined successfully!"
+    return refined, "Prompt refined successfully!"
 
 def create_interface():
     # Preload models if needed
@@ -225,15 +230,13 @@ def create_interface():
         with gr.Column(elem_id="col-container"):
             gr.Markdown("# Text to Product\nUsing Mistral-7B-Instruct-v0.3 + FLUX.1-dev + Trellis")
             
-            # Basic inputs
-            with gr.Row():
-                prompt = gr.Text(
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter basic object prompt",
-                    container=False,
-                )
-                prompt_button = gr.Button("Refine prompt with Mistral")
+            prompt = gr.Text(
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter basic object prompt",
+                container=False,
+            )
+            prompt_button = gr.Button("Refine prompt with Mistral")
             
             refined_prompt = gr.Text(
                 show_label=False,
@@ -242,10 +245,12 @@ def create_interface():
                 container=False,
                 max_length=2048,
             )
-            
             visual_button = gr.Button("Create visual with Flux")
+
             generated_image = gr.Image(show_label=False)
-            error_box = gr.Textbox(
+            gen3d_button = gr.Button("Create 3D visual with Trellis")
+
+            message_box = gr.Textbox(
                 label="Status Messages", 
                 interactive=False,
                 placeholder="Status messages will appear here",
@@ -293,7 +298,7 @@ def create_interface():
                 examples=examples,  # Now just a list of prompts
                 fn=process_example_pipeline,
                 inputs=[prompt],  # Add system_prompt as input
-                outputs=[prompt, refined_prompt, error_box],  # Don't output image
+                outputs=[refined_prompt, message_box],  # Don't output image
                 cache_examples=True,
             )
 
@@ -302,14 +307,14 @@ def create_interface():
             triggers=[prompt_button.click, prompt.submit],
             fn=refine_prompt,
             inputs=[prompt, system_prompt],  # Add system_prompt as input
-            outputs=[refined_prompt, error_box]
+            outputs=[refined_prompt, message_box]
         )
 
         gr.on(
             triggers=[visual_button.click],
             fn=infer,
             inputs=[refined_prompt, seed, randomize_seed, width, height, num_inference_steps],
-            outputs=[generated_image, error_box]
+            outputs=[generated_image, message_box]
         )
 
     return demo