AIRider commited on
Commit
22bef0a
·
verified ·
1 Parent(s): 12d804b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -20
app.py CHANGED
@@ -126,16 +126,19 @@ def fetch_crawl_results(query):
126
  references = fetch_references(query)
127
  return references[0], references[1], references[2]
128
 
129
- def generate_blog_post(query, prompt_template, temperature, top_p, frequency_penalty, presence_penalty):
130
  try:
131
  # 참고글 크롤링
132
  references = fetch_references(query)
133
  ref1, ref2, ref3 = references
134
 
135
  combined_content = f"참고글1:\n{ref1}\n\n참고글2:\n{ref2}\n\n참고글3:\n{ref3}"
 
 
 
136
 
137
- full_prompt = f"주제: {query}\n\n참고 내용:\n{combined_content}"
138
-
139
  response = openai.ChatCompletion.create(
140
  model="gpt-4o-mini",
141
  messages=[
@@ -143,17 +146,14 @@ def generate_blog_post(query, prompt_template, temperature, top_p, frequency_pen
143
  {"role": "user", "content": full_prompt}
144
  ],
145
  max_tokens=10000,
146
- temperature=temperature,
147
- top_p=top_p,
148
- frequency_penalty=frequency_penalty,
149
- presence_penalty=presence_penalty,
150
- seed=random.randint(1, 2**32 - 1)
151
  )
152
 
153
- generated_content = response.choices[0].message['content']
154
- final_content = f"주제: {query}\n\n{generated_content}"
155
-
156
- return final_content, ref1, ref2, ref3
157
  except Exception as e:
158
  return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
159
 
@@ -259,14 +259,7 @@ with gr.Blocks() as iface:
259
 
260
  generate_button.click(
261
  generate_blog_post,
262
- inputs=[
263
- query_input,
264
- prompt_input,
265
- temperature_input,
266
- top_p_input,
267
- frequency_penalty_input,
268
- presence_penalty_input
269
- ],
270
  outputs=[output_text, ref1_text, ref2_text, ref3_text],
271
  show_progress=True
272
  )
 
126
  references = fetch_references(query)
127
  return references[0], references[1], references[2]
128
 
129
+ def generate_blog_post(query, prompt_template):
130
  try:
131
  # 참고글 크롤링
132
  references = fetch_references(query)
133
  ref1, ref2, ref3 = references
134
 
135
  combined_content = f"참고글1:\n{ref1}\n\n참고글2:\n{ref2}\n\n참고글3:\n{ref3}"
136
+
137
+ # 랜덤 시드 생성
138
+ random_seed = random.randint(1, 100000)
139
 
140
+ full_prompt = f"주제: {query}\n\n{prompt_template}\n\n참고 내용:\n{combined_content}"
141
+
142
  response = openai.ChatCompletion.create(
143
  model="gpt-4o-mini",
144
  messages=[
 
146
  {"role": "user", "content": full_prompt}
147
  ],
148
  max_tokens=10000,
149
+ temperature=0.75,
150
+ top_p=1.0,
151
+ frequency_penalty=0.5,
152
+ presence_penalty=0.3,
153
+ seed=random_seed
154
  )
155
 
156
+ return f"주제: {query}\n\n{response.choices[0].message['content']}", ref1, ref2, ref3
 
 
 
157
  except Exception as e:
158
  return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
159
 
 
259
 
260
  generate_button.click(
261
  generate_blog_post,
262
+ inputs=[query_input, prompt_input],
 
 
 
 
 
 
 
263
  outputs=[output_text, ref1_text, ref2_text, ref3_text],
264
  show_progress=True
265
  )