Create app.py

app.py

@@ -0,0 +1,245 @@
+# AIPPT 服务替换方案：修订版
+
+根据您提供的 PPT 结构示例和 AIPPT 基本原理文档，我发现之前提供的提示词确实需要调整，以确保生成的数据结构与 PPTist 期望的格式完全匹配。以下是修订后的 Flask 实现方案，重点调整了提示词部分：
+
+## 修订后的 Flask 实现代码
+
+```python
+from flask import Flask, request, Response, stream_with_context
+import requests
+import os
+import json
+
+app = Flask(__name__)
+
+# 从环境变量获取配置
+OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
+OPENAI_API_URL = os.environ.get('OPENAI_API_URL', 'https://api.openai.com/v1/chat/completions')
+
+# PPT 大纲生成接口
+@app.route('/tools/aippt_outline', methods=['POST'])
+def aippt_outline():
+    data = request.json
+    content = data.get('content')
+    language = data.get('language', 'zh')
+    
+    # 构建发送给 OpenAI 的提示
+    messages = [
+        {
+            "role": "system",
+            "content": f"""您是一位专业的PPT大纲生成器。请为主题'{content}'创建一个详细的演示文稿大纲，使用{language}语言。
+            
+大纲应当遵循以下结构：
+# [主标题]
+## [一级章节标题]
+### [二级标题]
+- [要点1]
+- [要点2]
+...
+
+请确保大纲层次分明，内容全面且有深度。大纲应当包含3-6个一级章节，每个章节下有2-4个二级标题，每个二级标题包含2-4个要点。"""
+        }
+    ]
+    
+    def generate():
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {OPENAI_API_KEY}'
+        }
+        payload = {
+            'model': 'gpt-4o',
+            'messages': messages,
+            'stream': True
+        }
+        
+        response = requests.post(
+            OPENAI_API_URL,
+            headers=headers,
+            json=payload,
+            stream=True
+        )
+        
+        for line in response.iter_lines():
+            if line:
+                line = line.decode('utf-8')
+                if line.startswith('data: ') and line != 'data: [DONE]':
+                    try:
+                        data = json.loads(line[6:])
+                        if data['choices'][0]['delta'].get('content'):
+                            content = data['choices'][0]['delta']['content']
+                            yield content
+                    except Exception as e:
+                        print(f"Error parsing OpenAI response: {e}")
+    
+    return Response(stream_with_context(generate()), mimetype='text/event-stream')
+
+# PPT 内容生成接口
+@app.route('/tools/aippt', methods=['POST'])
+def aippt():
+    data = request.json
+    content = data.get('content')
+    language = data.get('language', 'zh')
+    
+    # 构建发送给 OpenAI 的提示，包含 PPTist 所需数据结构的详细说明
+    messages = [
+        {
+            "role": "system",
+            "content": f"""您是一位专业的PPT内容生成器。根据提供的大纲，生成符合AIPPT结构的JSON对象。每个幻灯片必须严格遵循以下格式，根据其类型包含适当的字段。
+
+以下是幻灯片类型及其必要字段格式：
+
+1. 封面页 (cover):
+```
+{{
+  "type": "cover",
+  "data": {{
+    "title": "标题文本",
+    "text": "副标题或描述文本"
+  }}
+}}
+```
+
+2. 目录页 (contents):
+```
+{{
+  "type": "contents",
+  "data": {{
+    "items": ["目录项1", "目录项2", "目录项3", ...]
+  }}
+}}
+```
+
+3. 过渡页 (transition):
+```
+{{
+  "type": "transition",
+  "data": {{
+    "title": "章节标题",
+    "text": "章节描述文本"
+  }}
+}}
+```
+
+4. 内容页 (content):
+```
+{{
+  "type": "content",
+  "data": {{
+    "title": "页面标题",
+    "items": [
+      {{
+        "title": "内容项标题1",
+        "text": "内容项文本1"
+      }},
+      {{
+        "title": "内容项标题2",
+        "text": "内容项文本2"
+      }},
+      ...
+    ]
+  }}
+}}
+```
+
+5. 结束页 (end):
+```
+{{
+  "type": "end"
+}}
+```
+
+请根据提供的大纲，按顺序生成以下幻灯片：
+1. 一个封面页，包含主题标题和简短描述
+2. 一个目录页，列出大纲中的所有一级章节
+3. 每个一级章节前添加一个过渡页
+4. 每个二级标题创建一个内容页，标题为二级标题名称，内容项为该标题下的要点
+5. 最后添加一个结束页
+
+每个幻灯片必须是一个完整有效的JSON对象，遵循上述格式。每个内容页应有2-4个内容项。
+
+请逐个输出每张幻灯片的JSON对象，每个对象单独成行，不要添加任何额外的文本说明。"""
+        },
+        {
+            "role": "user",
+            "content": f"根据这个大纲生成PPT内容：\n{content}"
+        }
+    ]
+    
+    def generate():
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {OPENAI_API_KEY}'
+        }
+        payload = {
+            'model': 'gpt-4o',
+            'messages': messages,
+            'stream': True
+        }
+        
+        response = requests.post(
+            OPENAI_API_URL,
+            headers=headers,
+            json=payload,
+            stream=True
+        )
+        
+        buffer = ""
+        for line in response.iter_lines():
+            if line:
+                line = line.decode('utf-8')
+                if line.startswith('data: ') and line != 'data: [DONE]':
+                    try:
+                        data = json.loads(line[6:])
+                        if data['choices'][0]['delta'].get('content'):
+                            buffer += data['choices'][0]['delta']['content']
+                            
+                            # 尝试从缓冲区中提取完整的 JSON 对象
+                            while True:
+                                # 寻找可能的 JSON 对象开始和结束
+                                json_start = buffer.find('{')
+                                if json_start == -1:
+                                    break
+                                
+                                # 寻找匹配的结束括号（处理嵌套JSON）
+                                bracket_count = 0
+                                json_end = -1
+                                
+                                for i in range(json_start, len(buffer)):
+                                    if buffer[i] == '{':
+                                        bracket_count += 1
+                                    elif buffer[i] == '}':
+                                        bracket_count -= 1
+                                        if bracket_count == 0:
+                                            json_end = i
+                                            break
+                                
+                                if json_end == -1:
+                                    # 没有找到匹配的结束括号，等待更多数据
+                                    break
+                                
+                                json_str = buffer[json_start:json_end+1]
+                                
+                                try:
+                                    # 验证是否为有效的 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'):
+                                        # 发送到客户端
+                                        yield json.dumps(slide_data) + '\n'
+                                        # 从缓冲区中移除已处理的部分
+                                        buffer = buffer[json_end+1:]
+                                    else:
+                                        # JSON格式不符合期望，从缓冲区删除并继续
+                                        buffer = buffer[json_end+1:]
+                                except Exception as e:
+                                    # 解析错误，可能不是完整的JSON，继续等待
+                                    print(f"Error parsing JSON: {e}")
+                                    break
+                    except Exception as e:
+                        print(f"Error parsing OpenAI response: {e}")
+    
+    return Response(stream_with_context(generate()), mimetype='text/event-stream')
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=7860)