From 90e496298b21f0594b85535ec47d26d2d7d5a9ed Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期四, 03 七月 2025 17:33:09 +0800
Subject: [PATCH] fix:生成的内容

---
 test2.py |  381 +++++++++++++++++++++++++++++++-----------------------
 1 files changed, 219 insertions(+), 162 deletions(-)

diff --git a/test2.py b/test2.py
index aec0dcb..d8105ab 100644
--- a/test2.py
+++ b/test2.py
@@ -17,7 +17,8 @@
 
 def generate_xiaohongshu_article(film_name, locations):
     """生成小红书风格的文章"""
-    prompt = f"""请为电影《{film_name}》的拍摄地写一篇小红书风格的文章。要求:
+    try:
+        prompt = f"""请为电影《{film_name}》的拍摄地写一篇小红书风格的文章。要求:
 1. 标题要吸引人,包含电影名称和拍摄地
 2. 开头要吸引人,可以用电影中的经典台词或场景引入
 3. 详细介绍每个拍摄地点的特色和亮点
@@ -28,19 +29,23 @@
 
 请以html格式返回,包含标题、正文和标签.生成内容字体要符合在小程序上合适的大小阅读,标题不要超过h3。"""
 
-    completion = client.chat.completions.create(
-        model="bot-20250512103613-6rwj8",
-        messages=[
-            {"role": "system", "content": "你是一个专业的小红书文案写手,擅长写吸引人的旅游攻略。"},
-            {"role": "user", "content": prompt},
-        ],
-    )
-    
-    return completion.choices[0].message.content
+        completion = client.chat.completions.create(
+            model="bot-20250512103613-6rwj8",
+            messages=[
+                {"role": "system", "content": "你是一个专业的小红书文案写手,擅长写吸引人的旅游攻略。"},
+                {"role": "user", "content": prompt},
+            ],
+        )
+        
+        return completion.choices[0].message.content
+    except Exception as e:
+        print(f"生成小红书文章失败: {str(e)}")
+        return ""
 
 def generate_travel_route(film_name, locations):
     """生成游玩路线"""
-    prompt = f"""请为电影《{film_name}》的拍摄地设计一条最优游玩路线。要求:
+    try:
+        prompt = f"""请为电影《{film_name}》的拍摄地设计一条最优游玩路线。要求:
 1. 考虑各个景点之间的距离和交通方式
 2. 合理安排游览顺序,避免来回奔波
 3. 预估每个景点的游览时间
@@ -68,26 +73,30 @@
     ]
 }}"""
 
-    completion = client.chat.completions.create(
-        model="bot-20250512103613-6rwj8",
-        messages=[
-            {"role": "system", "content": "你是一个专业的旅游路线规划师,擅长设计最优游览路线。"},
-            {"role": "user", "content": prompt},
-        ],
-    )
-    
-    try:
-        response_text = completion.choices[0].message.content
-        cleaned_text = re.sub(r'```json|```', '', response_text)
-        cleaned_text = re.sub(r'^[^{[]*', '', cleaned_text)
-        return json.loads(cleaned_text)
+        completion = client.chat.completions.create(
+            model="bot-20250512103613-6rwj8",
+            messages=[
+                {"role": "system", "content": "你是一个专业的旅游路线规划师,擅长设计最优游览路线。"},
+                {"role": "user", "content": prompt},
+            ],
+        )
+        
+        try:
+            response_text = completion.choices[0].message.content
+            cleaned_text = re.sub(r'```json|```', '', response_text)
+            cleaned_text = re.sub(r'^[^{[]*', '', cleaned_text)
+            return json.loads(cleaned_text)
+        except Exception as e:
+            print(f"解析路线失败: {str(e)}\n原始响应: {response_text}")
+            return None
     except Exception as e:
-        print(f"解析路线失败: {str(e)}\n原始响应: {response_text}")
+        print(f"生成游玩路线失败: {str(e)}")
         return None
 
 def generate_xiaohongshu_route(film_name, locations):
     """生成小红书风格的路线攻略"""
-    prompt = f"""请为电影《{film_name}》的拍摄地写一篇小红书风格的路线攻略。要求:
+    try:
+        prompt = f"""请为电影《{film_name}》的拍摄地写一篇小红书风格的路线攻略。要求:
 1. 标题要吸引人,突出"最佳路线"或"完美行程"等关键词
 2. 开头要说明这条路线是如何规划的,为什么这样安排
 3. 详细介绍每个景点的游览时间和交通方式
@@ -99,27 +108,37 @@
 
 请以html格式返回,包含标题、正文和标签,生成内容字体要符合在小程序上合适的大小阅读,标题不要超过h3。"""
 
-    completion = client.chat.completions.create(
-        model="bot-20250512103613-6rwj8",
-        messages=[
-            {"role": "system", "content": "你是一个专业的小红书文案写手,擅长写吸引人的旅游路线攻略。"},
-            {"role": "user", "content": prompt},
-        ],
-    )
-    
-    return completion.choices[0].message.content
+        completion = client.chat.completions.create(
+            model="bot-20250512103613-6rwj8",
+            messages=[
+                {"role": "system", "content": "你是一个专业的小红书文案写手,擅长写吸引人的旅游路线攻略。"},
+                {"role": "user", "content": prompt},
+            ],
+        )
+        
+        return completion.choices[0].message.content
+    except Exception as e:
+        print(f"生成小红书路线攻略失败: {str(e)}")
+        return ""
 
 def get_film_works():
-    url = "http://192.168.1.213:8090/flower/api/filmWorks/pending/create"
-    response = requests.get(url)
-    if response.status_code == 200:
-        return response.json().get("data", [])
-    return []
+    try:
+        url = "http://192.168.1.213:8090/flower/api/filmWorks/pending/create"
+        response = requests.get(url)
+        if response.status_code == 200:
+            return response.json().get("data", [])
+        else:
+            print(f"获取电影作品失败,状态码: {response.status_code}")
+            return []
+    except Exception as e:
+        print(f"获取电影作品失败: {str(e)}")
+        return []
 
 
 def get_location_info_from_model(film_name):
     """使用模型获取多个拍摄地点信息"""
-    prompt = f"""请为电影《{film_name}》生成所有主要拍摄地点的详细信息。每部电影通常有多个拍摄地点,请尽可能详细地列出所有重要的拍摄地点。
+    try:
+        prompt = f"""请为电影《{film_name}》生成所有主要拍摄地点的详细信息。每部电影通常有多个拍摄地点,请尽可能详细地列出所有重要的拍摄地点。
 
 对于每个拍摄地点,请提供以下信息:
 1. 拍摄地点名称
@@ -156,165 +175,203 @@
 ]
 
 请确保返回所有重要的拍摄地点,每个地点都要包含完整的信息。"""
-    # Non-streaming:
-    print("----- standard request -----")
-    completion = client.chat.completions.create(
-        model="bot-20250512103613-6rwj8",  # bot-20250512103613-6rwj8 为您当前的智能体的ID,注意此处与Chat API存在差异。差异对比详见 SDK使用指南
-        # messages=[
-        #     {"role": "system", "content": "你是DeepSeek,是一个 AI 人工智能助手"},
-        #     {"role": "user", "content": "常见的十字花科植物有哪些?"},
-        # ],
-        messages=[
-            {"role": "system",
-             "content": "你是一个专业的影视拍摄地点信息专家,请根据电影名称生成所有重要拍摄地点的详细信息。"},
-            {"role": "user", "content": prompt},
-        ],
-    )
-    print(completion.choices[0].message.content)
-    if hasattr(completion, "references"):
-        print(completion.references)
-    if hasattr(completion.choices[0].message, "reasoning_content"):
-        print(completion.choices[0].message.reasoning_content)  # 对于R1模型,输出reasoning content
-    # try:
-    #     # 尝试解析模型返回的JSON
-    #     response_text = completion.choices[0].message.content
-    #     location_info_list = json.loads(response_text)
-    #     return location_info_list
-    # except:
-    #     # 如果解析失败,返回默认值
-    #     return None
-        try:
-            response_text = completion.choices[0].message.content
-            # 清理响应内容
-            cleaned_text = re.sub(r'```json|```', '', response_text)  # 移除Markdown标记
-            cleaned_text = re.sub(r'^[^{[]*', '', cleaned_text)  # 清理开头非JSON内容
+        # Non-streaming:
+        print("----- standard request -----")
+        completion = client.chat.completions.create(
+            model="bot-20250512103613-6rwj8",  # bot-20250512103613-6rwj8 为您当前的智能体的ID,注意此处与Chat API存在差异。差异对比详见 SDK使用指南
+            # messages=[
+            #     {"role": "system", "content": "你是DeepSeek,是一个 AI 人工智能助手"},
+            #     {"role": "user", "content": "常见的十字花科植物有哪些?"},
+            # ],
+            messages=[
+                {"role": "system",
+                 "content": "你是一个专业的影视拍摄地点信息专家,请根据电影名称生成所有重要拍摄地点的详细信息。"},
+                {"role": "user", "content": prompt},
+            ],
+        )
+        print(completion.choices[0].message.content)
+        if hasattr(completion, "references"):
+            print(completion.references)
+        if hasattr(completion.choices[0].message, "reasoning_content"):
+            print(completion.choices[0].message.reasoning_content)  # 对于R1模型,输出reasoning content
+        # try:
+        #     # 尝试解析模型返回的JSON
+        #     response_text = completion.choices[0].message.content
+        #     location_info_list = json.loads(response_text)
+        #     return location_info_list
+        # except:
+        #     # 如果解析失败,返回默认值
+        #     return None
+            try:
+                response_text = completion.choices[0].message.content
+                # 清理响应内容
+                cleaned_text = re.sub(r'```json|```', '', response_text)  # 移除Markdown标记
+                cleaned_text = re.sub(r'^[^{[]*', '', cleaned_text)  # 清理开头非JSON内容
 
-            # 解析
-            location_info_list = json.loads(cleaned_text)
-            return location_info_list
-        except Exception as e:
-            print(f"解析失败: {str(e)}\n原始响应: {response_text}")
-            return None
+                # 解析
+                location_info_list = json.loads(cleaned_text)
+                return location_info_list
+            except Exception as e:
+                print(f"解析失败: {str(e)}\n原始响应: {response_text}")
+                return None
+    except Exception as e:
+        print(f"获取拍摄地点信息失败: {str(e)}")
+        return None
 
 
 def create_film_location(film_id, film_name, location_info, article, route_article):
-    url = "http://192.168.1.213:8090/flower/api/filmLocation/new"
+    try:
+        url = "http://192.168.1.213:8090/flower/api/filmLocation/new"
 
-    # 默认值设置
-    default_data = {
-        "filmId": film_id,
-        "locationName": f"{film_name}拍摄地",
-        "address": "待补充",
-        "gpsLat": 0.0,
-        "gpsLng": 0.0,
-        "startDate": datetime.now().strftime("%Y-%m-%d"),
-        "endDate": datetime.now().strftime("%Y-%m-%d"),
-        "sceneType": "外景",
-        "classicScene": "待补充",
-        "isOpenVisit": 0,
-        "visitInfo": json.dumps({"tips": "暂无参观信息"}),
-        "landmarkDesc": "待补充",
-        "transportGuide": "待补充",
-        "parkingInfo": "待补充",
-        "surroundingFacilities": "待补充",
-        "arEntry": "",
-        "status": 1,
-        "deleted": 0,
-        "createBy": "system",
-        "updateBy": "system",
-        "checkinCount": 0,
-        "visitorPhotos": json.dumps([]),
-        "xiaohongshuArticle": article,  # 新增:小红书风格文章
-        "xiaohongshuRoute": route_article  # 新增:小红书风格路线
-    }
+        # 默认值设置
+        default_data = {
+            "filmId": film_id,
+            "locationName": f"{film_name}拍摄地",
+            "address": "待补充",
+            "gpsLat": 0.0,
+            "gpsLng": 0.0,
+            "startDate": datetime.now().strftime("%Y-%m-%d"),
+            "endDate": datetime.now().strftime("%Y-%m-%d"),
+            "sceneType": "外景",
+            "classicScene": "待补充",
+            "isOpenVisit": 0,
+            "visitInfo": json.dumps({"tips": "暂无参观信息"}),
+            "landmarkDesc": "待补充",
+            "transportGuide": "待补充",
+            "parkingInfo": "待补充",
+            "surroundingFacilities": "待补充",
+            "arEntry": "",
+            "status": 1,
+            "deleted": 0,
+            "createBy": "system",
+            "updateBy": "system",
+            "checkinCount": 0,
+            "visitorPhotos": json.dumps([]),
+            "xiaohongshuArticle": article,  # 新增:小红书风格文章
+            "xiaohongshuRoute": route_article  # 新增:小红书风格路线
+        }
 
-    # 更新默认值
-    if location_info:
-        for key, value in location_info.items():
-            if key in default_data:
-                if key in ['visit_info', 'visitor_photos']:
-                    default_data[key] = json.dumps(value)
-                else:
-                    default_data[key] = value
+        # 更新默认值
+        if location_info:
+            for key, value in location_info.items():
+                if key in default_data:
+                    if key in ['visit_info', 'visitor_photos']:
+                        default_data[key] = json.dumps(value)
+                    else:
+                        default_data[key] = value
 
-    response = requests.post(url, json=default_data)
-    return response.json()
+        response = requests.post(url, json=default_data)
+        return response.json()
+    except Exception as e:
+        print(f"创建拍摄地点失败: {str(e)}")
+        return {"error": str(e)}
 
 def save_article_and_route(film_id, film_name, article, route, route_article):
     """保存文章和路线到文件"""
-    # 创建输出目录
-    output_dir = "output"
-    if not os.path.exists(output_dir):
-        os.makedirs(output_dir)
-    
-    # 保存文章
-    article_file = os.path.join(output_dir, f"{film_name}_article.md")
-    with open(article_file, "w", encoding="utf-8") as f:
-        f.write(article)
-    
-    # 保存路线(JSON格式)
-    route_file = os.path.join(output_dir, f"{film_name}_route.json")
-    with open(route_file, "w", encoding="utf-8") as f:
-        json.dump(route, f, ensure_ascii=False, indent=2)
-    
-    # 保存路线(小红书风格)
-    route_article_file = os.path.join(output_dir, f"{film_name}_route_article.md")
-    with open(route_article_file, "w", encoding="utf-8") as f:
-        f.write(route_article)
+    try:
+        # 创建输出目录
+        output_dir = "output"
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
+        
+        # 保存文章
+        article_file = os.path.join(output_dir, f"{film_name}_article.md")
+        with open(article_file, "w", encoding="utf-8") as f:
+            f.write(article if article else "文章生成失败")
+        
+        # 保存路线(JSON格式)
+        route_file = os.path.join(output_dir, f"{film_name}_route.json")
+        with open(route_file, "w", encoding="utf-8") as f:
+            if route:
+                json.dump(route, f, ensure_ascii=False, indent=2)
+            else:
+                json.dump({"error": "路线生成失败"}, f, ensure_ascii=False, indent=2)
+        
+        # 保存路线(小红书风格)
+        route_article_file = os.path.join(output_dir, f"{film_name}_route_article.md")
+        with open(route_article_file, "w", encoding="utf-8") as f:
+            f.write(route_article if route_article else "路线文章生成失败")
+    except Exception as e:
+        print(f"保存文件失败: {str(e)}")
 
 def main():
     # 获取所有电影作品
-    film_works = get_film_works()
+    try:
+        film_works = get_film_works()
+    except Exception as e:
+        print(f"获取电影作品失败: {str(e)}")
+        return
 
     # 为每个电影作品创建拍摄地点
     for film in film_works:
-        film_name = film.get("nameCn")
-        film_id = film.get("id")
-        if film_name:
-            print(f"正在处理电影: {film_name}")
+        try:
+            film_name = film.get("nameCn")
+            film_id = film.get("id")
+            if film_name:
+                print(f"正在处理电影: {film_name}")
 
-            # 获取所有拍摄地点信息
-            location_info_list = get_location_info_from_model(film_name)
+                # 获取所有拍摄地点信息
+                location_info_list = get_location_info_from_model(film_name)
 
-            if location_info_list:
                 # 生成小红书文章
                 article = generate_xiaohongshu_article(film_name, location_info_list)
                 # 清理文章中的HTML标记
-                article = re.sub(r'```html|```', '', article)
-                print(f"\n生成的文章:\n{article}")
+                if article:
+                    article = re.sub(r'```html|```', '', article)
+                    print(f"\n生成的文章:\n{article}")
+                else:
+                    print("文章生成失败,使用空内容")
                 
                 # 生成游玩路线(JSON格式)
                 route = generate_travel_route(film_name, location_info_list)
-                print(f"\n生成的路线:\n{json.dumps(route, ensure_ascii=False, indent=2)}")
+                if route:
+                    print(f"\n生成的路线:\n{json.dumps(route, ensure_ascii=False, indent=2)}")
+                else:
+                    print("路线生成失败,使用空内容")
                 
                 # 生成小红书风格路线
                 route_article = generate_xiaohongshu_route(film_name, location_info_list)
                 # 清理路线文章中的HTML标记
-                route_article = re.sub(r'```html|```', '', route_article)
-                print(f"\n生成的路线文章:\n{route_article}")
+                if route_article:
+                    route_article = re.sub(r'```html|```', '', route_article)
+                    print(f"\n生成的路线文章:\n{route_article}")
+                else:
+                    print("路线文章生成失败,使用空内容")
                 
                 # 合并文章和路线
                 combined_content = f"{article}\n\n{route_article}"
                 
                 # 保存到新接口
-                save_url = "http://192.168.1.213:8090/flower/api/filmWorks/edit"
-                save_data = {
-                    "id": film_id,
-                    "type": "sys",
-                    "filmContent": combined_content
-                }
-                save_response = requests.post(save_url, json=save_data)
-                print(f"保存到新接口结果: {save_response.json()}")
+                try:
+                    save_url = "http://192.168.1.213:8090/flower/api/filmWorks/edit"
+                    save_data = {
+                        "id": film_id,
+                        "type": "sys",
+                        "filmContent": combined_content
+                    }
+                    save_response = requests.post(save_url, json=save_data)
+                    print(f"保存到新接口结果: {save_response.json()}")
+                except Exception as e:
+                    print(f"保存到新接口失败: {str(e)}")
                 
                 # 为每个拍摄地点创建记录
-                for location_info in location_info_list:
-                    result = create_film_location(film_id, film_name, location_info, article, route_article)
-                    print(f"创建拍摄地点 {location_info.get('locationName', '未知地点')} 结果: {result}")
+                if location_info_list:
+                    for location_info in location_info_list:
+                        result = create_film_location(film_id, film_name, location_info, article, route_article)
+                        print(f"创建拍摄地点 {location_info.get('locationName', '未知地点')} 结果: {result}")
+                else:
+                    print("没有拍摄地点信息,跳过创建拍摄地点记录")
                 
                 # 保存文章和路线到文件
-                save_article_and_route(film_id, film_name, article, route, route_article)
+                try:
+                    save_article_and_route(film_id, film_name, article, route, route_article)
+                except Exception as e:
+                    print(f"保存文件失败: {str(e)}")
             else:
-                print(f"未能获取到电影 {film_name} 的拍摄地点信息")
+                print(f"电影名称为空,跳过处理")
+        except Exception as e:
+            print(f"处理电影 {film.get('nameCn', '未知电影')} 时发生错误: {str(e)}")
+            continue
 
 
 if __name__ == "__main__":

--
Gitblit v1.9.3