Update app.py

app.py

@@ -2,6 +2,10 @@ from flask import Flask, request, Response, stream_with_context
 import requests
 import os
 import json
+import urllib3
+
+# 禁用 SSL 警告
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 app = Flask(__name__)
 
@@ -15,6 +19,7 @@ def aippt_outline():
     data = request.json
     content = data.get('content')
     language = data.get('language', 'zh')
+    model = data.get('model', 'gpt-4o')
     
     # 构建发送给 OpenAI 的提示
     messages = [
@@ -49,7 +54,7 @@ def aippt_outline():
             'Authorization': f'Bearer {OPENAI_API_KEY}'
         }
         payload = {
-            'model': 'gpt-4.1-mini',
+            'model': model,
             'messages': messages,
             'stream': True
         }
@@ -82,6 +87,7 @@ def aippt():
     data = request.json
     content = data.get('content')
     language = data.get('language', 'zh')
+    model = data.get('model', 'gpt-4.1-mini')
     
     # 构建发送给 OpenAI 的提示，包含 PPTist 所需数据结构的详细说明
     messages = [
@@ -94,25 +100,25 @@ def aippt():
 2. 必须严格遵循以下JSON格式，任何格式错误都会导致幻灯片无法显示：
 
 封面页:
-{"type":"cover","data":{"title":"主标题","text":"副标题或简短描述"}}
+{{"type":"cover","data":{{"title":"主标题","text":"副标题或简短描述"}}}}
 
 目录页:
-{"type":"contents","data":{"items":["一级章节1","一级章节2","一级章节3","一级章节4","一级章节5"]}}
+{{"type":"contents","data":{{"items":["一级章节1","一级章节2","一级章节3","一级章节4","一级章节5"]}}}}
 
 过渡页:
-{"type":"transition","data":{"title":"章节标题","text":"章节简介"}}
+{{"type":"transition","data":{{"title":"章节标题","text":"章节简介"}}}}
 
 内容页:
-{"type":"content","data":{"title":"二级标题","items":[{"title":"要点1标题","text":"要点1描述"},{"title":"要点2标题","text":"要点2描述"},{"title":"要点3标题","text":"要点3描述"},{"title":"要点4标题","text":"要点4描述"}]}}
+{{"type":"content","data":{{"title":"二级标题","items":[{{"title":"要点1标题","text":"要点1描述"}},{{"title":"要点2标题","text":"要点2描述"}},{{"title":"要点3标题","text":"要点3描述"}},{{"title":"要点4标题","text":"要点4描述"}}]}}}}
 
 结束页:
-{"type":"end"}
+{{"type":"end"}}
 
 3. 转换规则：
-   - 大纲中的主标题("# "开头)转换为封面页
-   - 所有一级章节("## "开头)组成目录页
+   - 大纲中的主标题("#"开头)转换为封面页
+   - 所有一级章节("##"开头)组成目录页
    - 每个一级章节前生成一个过渡页
-   - 每个二级标题("### "开头)生成一个内容页
+   - 每个二级标题("###"开头)生成一个内容页
    - 内容页中的items数组由该二级标题下的要点("-"开头)组成
    - 最后添加一个结束页
 
@@ -123,10 +129,9 @@ def aippt():
    - 每个内容页必须包含该节点下所有的要点，通常为2-4个
 
 示例输出：
-{"type":"cover","data":{"title":"5G技术如何改变我们的生活","text":"探索5G技术对现代生活的全方位影响"}}
-{"type":"contents","data":{"items":["5G技术概述","5G对通信领域的变革","5G与智能家居的融合","5G推动智能交通发展","5G在医疗领域的应用","5G助力工业互联网升级"]}}
-{"type":"transition","data":{"title":"5G技术概述","text":"本章将介绍5G技术的定义、关键特性及发展历程。"}}
-..."""
+{{"type":"cover","data":{{"title":"5G技术如何改变我们的生活","text":"探索5G技术对现代生活的全方位影响"}}}}
+{{"type":"contents","data":{{"items":["5G技术概述","5G对通信领域的变革","5G与智能家居的融合","5G推动智能交通发展","5G在医疗领域的应用","5G助力工业互联网升级"]}}}}
+{{"type":"transition","data":{{"title":"5G技术概述","text":"本章将介绍5G技术的定义、关键特性及发展历程。"}}}}"""
         },
         {
             "role": "user",
@@ -140,7 +145,7 @@ def aippt():
             'Authorization': f'Bearer {OPENAI_API_KEY}'
         }
         payload = {
-            'model': 'gpt-4o',
+            'model': model,
             'messages': messages,
             'stream': True
         }
@@ -193,17 +198,17 @@ def aippt():
                                     # 验证是否为有效的 JSON
                                     slide_data = json.loads(json_str)
                                     
-                                    # 如果是有效的JSON且包含type和data字段（为了安全起见）
-                                    if isinstance(slide_data, dict) and ('type' in slide_data or slide_data.get('type') == 'end'):
+                                    # 验证是否包含必要的字段
+                                    if isinstance(slide_data, dict) and 'type' in slide_data:
                                         # 发送到客户端
                                         yield json.dumps(slide_data) + '\n'
                                         # 从缓冲区中移除已处理的部分
                                         buffer = buffer[json_end+1:]
                                     else:
-                                        # JSON格式不符合期望，从缓冲区删除并继续
+                                        # 格式不符合要求，从缓冲区移除
                                         buffer = buffer[json_end+1:]
                                 except Exception as e:
-                                    # 解析错误，可能不是完整的JSON，继续等待
+                                    # 解析错误，可能不是完整的JSON，等待更多数据
                                     print(f"Error parsing JSON: {e}")
                                     break
                     except Exception as e:
@@ -211,9 +216,5 @@ def aippt():
     
     return Response(stream_with_context(generate()), mimetype='text/event-stream')
 
-# 添加请求警告过滤器，抑制 SSL 验证警告
-import urllib3
-urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
-
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)