From 65618b2c0b570d22dc6043c45d446e4801bfcdc0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A4=AA=E4=B8=80?= <2339117167@qq.com>
Date: Sun, 13 Apr 2025 18:30:59 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=8E=B7=E5=8F=96=E7=BD=91?=
 =?UTF-8?q?=E9=A1=B5=E5=86=85=E5=AE=B9=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ArticleReplaceBatch/main_process.py | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/ArticleReplaceBatch/main_process.py b/ArticleReplaceBatch/main_process.py
index 2874ab1..d800280 100644
--- a/ArticleReplaceBatch/main_process.py
+++ b/ArticleReplaceBatch/main_process.py
@@ -1,11 +1,9 @@
-import pandas as pd
-import getpass
-import sys  # 导入sys模块
+
 import threading
 import queue
 
 
-from ai_studio import call_dify_workflow
+from ai_studio import call_dify_workflow, call_coze_workflow
 from databases import *
 
 
@@ -15,9 +13,11 @@ from get_web_content import  *
 from config import *
 
 # ==============================主程序===========================
-def process_link(link):
+def process_link(link, ai_service="dify"):
     """
     处理单个链接
+    :param link: 要处理的链接
+    :param ai_service: AI服务提供商，可选值：dify, coze
     """
     try:
         if link.startswith("https://www.toutiao.com"):
@@ -63,7 +63,16 @@ def process_link(link):
             "old_article": article_text
         }
 
-        message_content = call_dify_workflow(input_data)
+        if ai_service == "dify":
+            input_data = {
+                "old_article": article_text
+            }
+            message_content = call_dify_workflow(input_data)
+        elif ai_service == "coze":
+            input_data = {
+                "old_article": article_text
+            }
+            message_content = call_coze_workflow(input_data)
         # 获取当前时间并格式化
         current_time = datetime.now().strftime("%H:%M:%S")
 
@@ -109,7 +118,7 @@ def process_link(link):
 
 
 
-def link_to_text(prompt1=None, prompt2=None, num_threads=None):
+def link_to_text(prompt1=None, prompt2=None, num_threads=None, ai_service="dify"):
     use_link_path = 'use_link_path.txt'
 
     # 读取链接
@@ -165,7 +174,7 @@ def worker():
             # 处理链接
             try:
                 logger.info(f"开始处理链接：{link}")
-                process_link(link)
+                process_link(link, ai_service)
                 result_queue.put((link, True, None))  # 成功
             except Exception as e:
                 result_queue.put((link, False, str(e)))  # 失败