yooke commited on
Commit
fe9cd1e
·
verified ·
1 Parent(s): ebe7546

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py CHANGED
@@ -50,17 +50,37 @@ class BasicAgent:
50
  # Ensure result is always a string
51
  if not isinstance(result, str):
52
  result = str(result)
 
 
 
 
 
53
  return result
54
  except Exception as e:
55
  print(traceback.format_exc())
56
  error_msg = f"Error running agent: {str(e)}"
57
  return f"I encountered an issue while processing your question: {str(e)}"
58
 
 
 
 
 
 
59
  def run_and_submit_all( profile: gr.OAuthProfile | None):
60
  """
61
  Fetches all questions, runs the BasicAgent on them, submits all answers,
62
  and displays the results.
63
  """
 
 
 
 
 
 
 
 
 
 
64
  # --- Determine HF Space Runtime URL and Repo URL ---
65
  space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
66
 
@@ -149,6 +169,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
149
  f"Message: {result_data.get('message', 'No message received.')}"
150
  )
151
  # print("Submission successful.")
 
152
  results_df = pd.DataFrame(results_log)
153
  return final_status, results_df
154
  except requests.exceptions.HTTPError as e:
 
50
  # Ensure result is always a string
51
  if not isinstance(result, str):
52
  result = str(result)
53
+ # 提取最终答案 - 如果结果包含"Final answer:"格式
54
+ if "Final answer:" in result:
55
+ # 只提取最终答案部分
56
+ final_answer_part = result.split("Final answer:")[1].strip()
57
+ return final_answer_part
58
  return result
59
  except Exception as e:
60
  print(traceback.format_exc())
61
  error_msg = f"Error running agent: {str(e)}"
62
  return f"I encountered an issue while processing your question: {str(e)}"
63
 
64
+
65
+
66
+ # 添加全局变量来存储已经处理过的问题和答案
67
+ processed_questions = {}
68
+ submission_completed = False
69
  def run_and_submit_all( profile: gr.OAuthProfile | None):
70
  """
71
  Fetches all questions, runs the BasicAgent on them, submits all answers,
72
  and displays the results.
73
  """
74
+
75
+ global processed_questions, submission_completed
76
+
77
+ # 如果已经完成提交,直接返回之前的结果
78
+ if submission_completed and processed_questions:
79
+ results_log = []
80
+ for task_id, data in processed_questions.items():
81
+ results_log.append({"Task ID": task_id, "Question": data["question"], "Submitted Answer": data["answer"]})
82
+ return "已经完成提交,请刷新页面重新开始。", pd.DataFrame(results_log)
83
+
84
  # --- Determine HF Space Runtime URL and Repo URL ---
85
  space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
86
 
 
169
  f"Message: {result_data.get('message', 'No message received.')}"
170
  )
171
  # print("Submission successful.")
172
+ submission_completed = True
173
  results_df = pd.DataFrame(results_log)
174
  return final_status, results_df
175
  except requests.exceptions.HTTPError as e: