Update app.py

app.py

@@ -39,7 +39,7 @@ def mistral_inference(prompt, image_url):
 # Gradio インターフェース
 def process_input(text, image_url):
     result = mistral_inference(text, image_url)
-    return result
+    return result, image_url
 
 with gr.Blocks() as demo:
     gr.Markdown("## Pixtralモデルによる画像説明生成")
@@ -48,10 +48,12 @@ with gr.Blocks() as demo:
         text_input = gr.Textbox(label="テキストプロンプト", placeholder="例: Describe the image.")
         image_input = gr.Textbox(label="画像URL", placeholder="例: https://example.com/image.png")
     
-    result_output = gr.Textbox(label="モデルの出力結果")
-    
+    result_output = gr.Textbox(label="モデルの出力結果", lines=8, max_lines=20)  # 高さを500ピクセルに相当するように調整
+    image_output = gr.Image(label="入力された画像", type="auto")  # 入力画像URLを表示するための場所
+
     submit_button = gr.Button("推論を実行")
-    
-    submit_button.click(process_input, inputs=[text_input, image_input], outputs=result_output)
+
+    # ボタンをクリックすると、モデルの結果と画像を表示
+    submit_button.click(process_input, inputs=[text_input, image_input], outputs=[result_output, image_output])
 
 demo.launch()