Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ import re
|
|
6 |
import uuid
|
7 |
import json
|
8 |
|
9 |
-
#
|
10 |
-
|
11 |
|
12 |
OUTPUT_DIR = "outputs"
|
13 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
@@ -19,8 +19,13 @@ def extract_task_from_jd(jd):
|
|
19 |
任务:(任务描述)
|
20 |
|
21 |
JD: {jd}"""
|
22 |
-
|
23 |
-
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
# 基于任务生成三个解决方案
|
26 |
def generate_solutions_from_task(task):
|
@@ -31,11 +36,19 @@ def generate_solutions_from_task(task):
|
|
31 |
方案3:(内容)
|
32 |
|
33 |
任务: {task}"""
|
34 |
-
|
35 |
-
|
36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
if len(solutions) < 3:
|
38 |
-
return
|
|
|
39 |
s1 = solutions[0].strip()
|
40 |
s2 = solutions[1].strip()
|
41 |
s3 = solutions[2].strip()
|
|
|
6 |
import uuid
|
7 |
import json
|
8 |
|
9 |
+
# 初始化 Qwen 客户端(Qwen API)
|
10 |
+
client = Client("https://qwen-qwen-72b-chat-demo.hf.space/--replicas/ypz66/")
|
11 |
|
12 |
OUTPUT_DIR = "outputs"
|
13 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
|
19 |
任务:(任务描述)
|
20 |
|
21 |
JD: {jd}"""
|
22 |
+
result = client.predict(
|
23 |
+
message, # 输入 JD 描述
|
24 |
+
[["Hello!", None]], # Chatbot 输入
|
25 |
+
"Hello!!", # 作为参数 9 的文本
|
26 |
+
api_name="/model_chat" # API 名称
|
27 |
+
)
|
28 |
+
return result.strip() # 直接返回原始响应内容,便于调试
|
29 |
|
30 |
# 基于任务生成三个解决方案
|
31 |
def generate_solutions_from_task(task):
|
|
|
36 |
方案3:(内容)
|
37 |
|
38 |
任务: {task}"""
|
39 |
+
result = client.predict(
|
40 |
+
message, # 输入任务描述
|
41 |
+
[["Hello!", None]], # Chatbot 输入
|
42 |
+
"Hello!!", # 作为参数 9 的文本
|
43 |
+
api_name="/model_chat" # API 名称
|
44 |
+
)
|
45 |
+
|
46 |
+
# 使用正则表达式提取方案
|
47 |
+
solutions = re.findall(r"方案[123][::]\s*(.*)", result)
|
48 |
+
|
49 |
if len(solutions) < 3:
|
50 |
+
return result.strip(), "(解析失败,显示原始回复)", ""
|
51 |
+
|
52 |
s1 = solutions[0].strip()
|
53 |
s2 = solutions[1].strip()
|
54 |
s3 = solutions[2].strip()
|