Update app.py

app.py

@@ -70,7 +70,7 @@ def infer(image_in):
     
     print(f"SUGGESTED LLM: {cleaned_text}")
     
-    return cleaned_text.lstrip("\n")
+    return user_prompt, cleaned_text.lstrip("\n")
 
 title = f"LLM Agent from a Picture",
 description = f"Get a LLM system prompt from a picture so you can use it in <a href='https://huggingface.co/spaces/abidlabs/GPT-Baker'>GPT-Baker</a>."
@@ -99,8 +99,12 @@ with gr.Blocks(css=css) as demo:
                 )
                 submit_btn = gr.Button("Make LLM system from my pic !")
             with gr.Column():
+                caption = gr.Textbox(
+                    label = "Image caption (fuyu)",
+                    elem_id = "image-caption"
+                )
                 result = gr.Textbox(
-                    label ="Suggested System",
+                    label = "Suggested System",
                     lines = 6,
                     max_lines = 30,
                     elem_id = "suggested-system-prompt"
@@ -116,7 +120,7 @@ with gr.Blocks(css=css) as demo:
                 ],
                 fn = infer,
                 inputs = [image_in],
-                outputs = [result],
+                outputs = [caption, result],
                 cache_examples = True
             )
 
@@ -126,6 +130,7 @@ with gr.Blocks(css=css) as demo:
             image_in
         ],
         outputs =[
+            caption,
             result
         ]
     )