Update app.py

app.py

@@ -122,14 +122,24 @@ def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence_
     if task_prompt == 'Product Caption':
         task_prompt = '<PC>'
         results = run_example(task_prompt, image, model_id=model_id)
-        return results, None
     elif task_prompt == 'More Detailed Caption':
         task_prompt = '<MORE_DETAILED_CAPTION>'
         results = run_example(task_prompt, image, model_id=model_id)
-        return results, None
     else:
         return "", None  # Return empty string and None for unknown task prompts
 
+    # Remove the key and get the text value
+    if results and task_prompt in results:
+        output_text = results[task_prompt]
+    else:
+        output_text = ""
+
+    # Convert newline characters to HTML line breaks
+    output_text = output_text.replace("\n\n", "<br><br>").replace("\n", "<br>")
+
+    return output_text, None
+
+
 css = """
   #output {
     height: 500px; 
@@ -157,9 +167,9 @@ with gr.Blocks(css=css) as demo:
                 text_input = gr.Textbox(label="Text Input (optional)")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column():
-                output_text = gr.Textbox(label="Output Text")
+                output_text = gr.HTML(label="Output Text")
                 output_img = gr.Image(label="Output Image")
 
         submit_btn.click(process_image, [input_img, task_prompt, text_input, model_selector], [output_text, output_img])
 
-demo.launch(debug=True)
+demo.launch(debug=True)