Upload content_generator.py with huggingface_hub
Browse files- 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.
|
| 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.
|
| 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.
|
| 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,
|