Heyyaha commited on
Commit
08abd4c
·
verified ·
1 Parent(s): 7683d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -8
app.py CHANGED
@@ -6,8 +6,8 @@ import re
6
  import uuid
7
  import json
8
 
9
- # 初始化任务生成客户端(Qwen API)
10
- taskgen_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,8 +19,13 @@ def extract_task_from_jd(jd):
19
  任务:(任务描述)
20
 
21
  JD: {jd}"""
22
- response = taskgen_client.predict(message=message, api_name="/model_chat")
23
- return response.strip() # 直接返回原始响应内容,便于调试
 
 
 
 
 
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
- response = taskgen_client.predict(message=message, api_name="/model_chat")
35
- # fallback:若正则提取失败,直接展示原始响应
36
- solutions = re.findall(r"方案[123][::]\s*(.*)", response)
 
 
 
 
 
 
 
37
  if len(solutions) < 3:
38
- return response.strip(), "(解析失败,显示原始回复)", ""
 
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()