Update app.py

app.py

@@ -99,7 +99,7 @@ def update_messages_with_text(messages, generated_content):
     messages.append(message_item)
     return messages
 
-def call_chatgpt_api(messages, client, max_tokens=10000, stop=None, temperature=1.1):
+def call_chatgpt_api(messages, client, max_tokens=10000, stop=None, temperature=1.0):
     """Call ChatGPT API with the given messages"""
     try:
         response = client.chat.completions.create(
@@ -254,7 +254,11 @@ def o3_chat(api_key, base_url, question, image):
     messages = evaluate_single_data(data, client, executor, prompt_template, prompt_type)
     print(messages)
     html_output = process_message(messages)
-    return html_output
+    
+    # 将消息转换为JSON字符串，用于下载
+    json_str = json.dumps(messages, ensure_ascii=False, indent=4)
+    
+    return html_output, json_str
 
 # Gradio界面
 def create_demo():
@@ -272,11 +276,37 @@ def create_demo():
             
         with gr.Row():
             output = gr.HTML(label="Response")
+            
+        # 添加JSON下载功能
+        json_output = gr.JSON(label="Messages JSON", visible=False)
+        download_btn = gr.Button("Download Messages as JSON")
         
-        submit_btn.click(
+        # 处理提交
+        result = submit_btn.click(
             fn=o3_chat,
             inputs=[api_key, base_url, question, image_input],
-            outputs=[output]
+            outputs=[output, json_output]
+        )
+        
+        # 添加下载功能
+        download_btn.click(
+            fn=lambda x: x,
+            inputs=[json_output],
+            outputs=[],
+            _js="""
+            function(json_str) {
+                const blob = new Blob([JSON.stringify(json_str, null, 2)], { type: 'application/json' });
+                const url = URL.createObjectURL(blob);
+                const a = document.createElement('a');
+                a.href = url;
+                a.download = 'messages_' + new Date().toISOString().replace(/[:.]/g, '-') + '.json';
+                document.body.appendChild(a);
+                a.click();
+                document.body.removeChild(a);
+                URL.revokeObjectURL(url);
+                return [];
+            }
+            """
         )
         
         gr.Markdown("""
@@ -294,3 +324,4 @@ if __name__ == "__main__":
     demo = create_demo()
     demo.launch()
 
+