AIRider commited on
Commit
2f65748
·
verified ·
1 Parent(s): 8c1d98e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -10
app.py CHANGED
@@ -134,28 +134,34 @@ def generate_blog_post(query, prompt_template):
134
  references = fetch_references(query)
135
  ref1, ref2, ref3 = references
136
 
137
- combined_content = f"참고글1:\n{ref1}\n\n참고글2:\n{ref2}\n\n참고글3:\n{ref3}"
138
-
139
  # 랜덤 시드 생성
140
  random_seed = random.randint(1, sys.maxsize)
141
 
142
- full_prompt = f"주제: {query}\n\n{prompt_template}\n\n참고 내용:\n{combined_content}"
 
 
 
 
 
 
 
143
 
 
144
  response = openai.ChatCompletion.create(
145
  model="gpt-4o-mini",
146
- messages=[
147
- {"role": "system", "content": prompt_template},
148
- {"role": "user", "content": f"주제: {query}\n\n참고 내용:\n{combined_content}"}
149
- ],
150
  max_tokens=10000,
151
- temperature=0.85,
152
- top_p=0.9,
153
  frequency_penalty=0.5,
154
  presence_penalty=0.3,
155
  seed=random_seed
156
  )
157
 
158
- return f"주제: {query}\n\n{response.choices[0].message['content']}", ref1, ref2, ref3
 
 
 
159
  except Exception as e:
160
  return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
161
 
 
134
  references = fetch_references(query)
135
  ref1, ref2, ref3 = references
136
 
 
 
137
  # 랜덤 시드 생성
138
  random_seed = random.randint(1, sys.maxsize)
139
 
140
+ # 메시지 구조 생성
141
+ messages = [
142
+ {"role": "system", "content": prompt_template},
143
+ {"role": "user", "content": f"주제: {query}"},
144
+ {"role": "user", "content": f"참고글1:\n{ref1}"},
145
+ {"role": "user", "content": f"참고글2:\n{ref2}"},
146
+ {"role": "user", "content": f"참고글3:\n{ref3}"}
147
+ ]
148
 
149
+ # OpenAI API 호출
150
  response = openai.ChatCompletion.create(
151
  model="gpt-4o-mini",
152
+ messages=messages,
 
 
 
153
  max_tokens=10000,
154
+ temperature=0.75,
155
+ top_p=1.0,
156
  frequency_penalty=0.5,
157
  presence_penalty=0.3,
158
  seed=random_seed
159
  )
160
 
161
+ # 생성된 블로그 포스트
162
+ generated_post = response.choices[0].message['content']
163
+
164
+ return generated_post, ref1, ref2, ref3
165
  except Exception as e:
166
  return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
167