Update app.py

app.py

@@ -390,7 +390,7 @@ def o3_chat(model_name, api_key, base_url, question, image):
 
     sharegpt_data_item = process_message_to_sharegpt_format(messages)
     
-    return html_output, sharegpt_data_item['images']
+    return html_output, sharegpt_data_item['images'], sharegpt_data_item['sharegpt_conversation']
 
 # Gradio界面
 def create_demo():
@@ -416,13 +416,16 @@ def create_demo():
             output = gr.HTML(label="Response")
 
         with gr.Row():
-            extracted_images = gr.Gallery(label="Extracted Images")
+            with gr.Column(scale=1):
+                extracted_images = gr.Gallery(label="Extracted Images")
+            with gr.Column(scale=1):
+                extracted_json = gr.JSON(label="Extracted JSON")
         
         # 处理提交
         submit_btn.click(
             fn=o3_chat,
             inputs=[model_name, api_key, base_url, question, image_input],
-            outputs=[output, extracted_images]
+            outputs=[output, extracted_images, extracted_json]
         )
         
         # 示例部分