Zekun Wu commited on
Commit
4efe666
·
1 Parent(s): 0bacd3a
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -45,6 +45,8 @@ class GPTAgent:
45
  return output
46
 
47
 
 
 
48
  def generate_potshot_prompt(batch, role, tone, audience, values):
49
  prompt = f"""You are a recruitment specialist tasked with generating witty and engaging recruitment potshots.
50
 
@@ -68,7 +70,8 @@ def get_potshots(n_repeat=1, batch=20, role="Developer", tone="humorous", audien
68
  while len(total_potshots) < desired_count:
69
  needed = min(batch, desired_count - len(total_potshots))
70
  prompt = generate_potshot_prompt(batch=needed, role=role, tone=tone, audience=audience, values=values)
71
- response = GPTAgent.invoke(prompt)
 
72
  try:
73
  batch_potshots = json.loads(response).get("potshots", [])
74
  total_potshots.extend(batch_potshots)
@@ -101,6 +104,8 @@ values = st.text_area("Company Values", "innovation, teamwork, transparency")
101
  batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10)
102
  repeat_times = st.number_input("Number of Batches", min_value=1, max_value=10, value=1)
103
 
 
 
104
  if st.button("Generate Potshots"):
105
  with st.spinner("Generating customized potshots..."):
106
  results = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,
 
45
  return output
46
 
47
 
48
+
49
+
50
  def generate_potshot_prompt(batch, role, tone, audience, values):
51
  prompt = f"""You are a recruitment specialist tasked with generating witty and engaging recruitment potshots.
52
 
 
70
  while len(total_potshots) < desired_count:
71
  needed = min(batch, desired_count - len(total_potshots))
72
  prompt = generate_potshot_prompt(batch=needed, role=role, tone=tone, audience=audience, values=values)
73
+ gpt_agent = GPTAgent()
74
+ response = gpt_agent.invoke(prompt)
75
  try:
76
  batch_potshots = json.loads(response).get("potshots", [])
77
  total_potshots.extend(batch_potshots)
 
104
  batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10)
105
  repeat_times = st.number_input("Number of Batches", min_value=1, max_value=10, value=1)
106
 
107
+
108
+
109
  if st.button("Generate Potshots"):
110
  with st.spinner("Generating customized potshots..."):
111
  results = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,