Update app.py

app.py

@@ -53,12 +53,43 @@ def infer(image_in):
 
     return cleaned_text
 
-gr.Interface(
-    fn = infer,
-    inputs = [
-        gr.Image(type="filepath")
-    ],
-    outputs = [
-        gr.Textbox()
-    ]
-).queue().launch()
+title = "LLM Agent from a Picture",
+description = "Get a LLM system prompt from a picture to then use in GPT-Baker."
+
+css = """
+#col-container{
+    margin: 0 auto;
+    max-width: 840px;
+    text-align: left;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.HTML(f"""
+        <h2 style="text-align: center;">{title}</h2>
+        <p style="text-align: center;">{description}</p>
+        """)
+    with gr.Row():
+        with gr.Column():
+            image_in = gr.Image(
+                label = "Image reference",
+                type = "filepath"
+            )
+            submit_btn = gr.Button("Make LLM system from my pic !")
+        with gr.Column():
+            result = gr.Textbox(
+                label ="Suggested System"
+            )
+
+    submit_btn.click(
+        fn = infer,
+        inputs = [
+            image_in
+        ],
+        outputs =[
+            result
+        ]
+    )
+
+demo.queue().launch()