FincH12 commited on
Commit
a3d8ff3
·
verified ·
1 Parent(s): 698348e

Upload content_generator.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. content_generator.py +3 -15
content_generator.py CHANGED
@@ -209,11 +209,7 @@ def generate_single_post(news_context, exclude_posts=None):
209
  settings = settings_manager.load_settings()
210
  prompt_template = settings.get("PROMPT_SINGLE", "")
211
 
212
- prompt = prompt_template.format(
213
- news_context=news_context,
214
- exclude_str=exclude_str,
215
- tone=tone
216
- )
217
 
218
  print("[Content Generator] Генерация одиночного поста (Persona Skills)...")
219
  raw_res = generate_text_via_gemini(prompt)
@@ -247,12 +243,7 @@ def generate_carousel_post(news_context, exclude_posts=None):
247
  prompt_template = settings.get("PROMPT_CAROUSEL", "")
248
  slide_count = settings.get("CAROUSEL_SLIDE_COUNT", 5)
249
 
250
- prompt = prompt_template.format(
251
- news_context=news_context,
252
- exclude_str=exclude_str,
253
- tone=tone,
254
- slide_count=slide_count
255
- )
256
  print("[Content Generator] Генерация поста-карусели (Persona Skills)...")
257
  raw_res = generate_text_via_gemini(prompt)
258
  data = clean_json_response(raw_res)
@@ -277,10 +268,7 @@ def predict_post_success(post_text, historical_data=None):
277
 
278
  settings = settings_manager.load_settings()
279
  prompt_template = settings.get("PROMPT_PREDICTOR", "")
280
- prompt = prompt_template.format(
281
- context=context,
282
- post_text=post_text
283
- )
284
  headers = {"Content-Type": "application/json"}
285
  payload = {
286
  "model": GEMINI_MODEL,
 
209
  settings = settings_manager.load_settings()
210
  prompt_template = settings.get("PROMPT_SINGLE", "")
211
 
212
+ prompt = prompt_template.replace("{news_context}", str(news_context)).replace("{exclude_str}", str(exclude_str)).replace("{tone}", str(tone))
 
 
 
 
213
 
214
  print("[Content Generator] Генерация одиночного поста (Persona Skills)...")
215
  raw_res = generate_text_via_gemini(prompt)
 
243
  prompt_template = settings.get("PROMPT_CAROUSEL", "")
244
  slide_count = settings.get("CAROUSEL_SLIDE_COUNT", 5)
245
 
246
+ prompt = prompt_template.replace("{news_context}", str(news_context)).replace("{exclude_str}", str(exclude_str)).replace("{tone}", str(tone)).replace("{slide_count}", str(slide_count))
 
 
 
 
 
247
  print("[Content Generator] Генерация поста-карусели (Persona Skills)...")
248
  raw_res = generate_text_via_gemini(prompt)
249
  data = clean_json_response(raw_res)
 
268
 
269
  settings = settings_manager.load_settings()
270
  prompt_template = settings.get("PROMPT_PREDICTOR", "")
271
+ prompt = prompt_template.replace("{context}", str(context)).replace("{post_text}", str(post_text))
 
 
 
272
  headers = {"Content-Type": "application/json"}
273
  payload = {
274
  "model": GEMINI_MODEL,