|
|
import os |
|
|
import requests |
|
|
import zeep |
|
|
from datetime import datetime, timedelta |
|
|
import telegram |
|
|
import time |
|
|
|
|
|
|
|
|
HF_TOKEN = os.getenv("HF_TOKEN") |
|
|
TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN") |
|
|
CHAT_ID = "-4620812399" |
|
|
|
|
|
|
|
|
wsdl_url = 'https://cbr.ru/DailyInfoWebServ/DailyInfo.asmx?WSDL' |
|
|
client = zeep.Client(wsdl=wsdl_url) |
|
|
|
|
|
def get_ruonia(start_date, end_date): |
|
|
start_str = start_date.strftime('%Y-%m-%d') |
|
|
end_str = end_date.strftime('%Y-%m-%d') |
|
|
try: |
|
|
data = client.service.Ruonia(fromDate=start_str, ToDate=end_str) |
|
|
return data |
|
|
except Exception as e: |
|
|
return [] |
|
|
|
|
|
def summarize_ruonia(data): |
|
|
summary = [] |
|
|
total_rate = 0 |
|
|
total_volume = 0 |
|
|
for entry in data: |
|
|
date = entry['D0'].strftime('%d.%m.%Y') |
|
|
rate = float(entry['RUONIA']) |
|
|
volume = float(entry['Volume']) |
|
|
total_rate += rate |
|
|
total_volume += volume |
|
|
summary.append(f"📅 {date}: Ставка RUONIA: {rate:.4f}%, Объём: {volume:.2f} млн руб.") |
|
|
avg_rate = total_rate / len(data) if data else 0 |
|
|
return "\n".join(summary), avg_rate, total_volume |
|
|
|
|
|
def get_weather(): |
|
|
url = "https://wttr.in/Moscow?format=3" |
|
|
try: |
|
|
res = requests.get(url) |
|
|
if res.status_code == 200: |
|
|
return res.text |
|
|
else: |
|
|
return "Погода в Москве: не удалось получить данные" |
|
|
except Exception: |
|
|
return "Погода в Москве: ошибка подключения" |
|
|
|
|
|
def ask_hf(prompt): |
|
|
headers = { |
|
|
"Authorization": f"Bearer {HF_TOKEN}", |
|
|
"Content-Type": "application/json" |
|
|
} |
|
|
payload = {"inputs": prompt} |
|
|
url = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta" |
|
|
try: |
|
|
res = requests.post(url, headers=headers, json=payload, timeout=60) |
|
|
output = res.json() |
|
|
if isinstance(output, list) and "generated_text" in output[0]: |
|
|
return output[0]["generated_text"] |
|
|
else: |
|
|
return "⚠️ Не удалось получить ответ нейросети" |
|
|
except Exception: |
|
|
return "⚠️ Ошибка запроса к HF API" |
|
|
|
|
|
def send_telegram_message(text): |
|
|
bot = telegram.Bot(token=TELEGRAM_TOKEN) |
|
|
bot.send_message(chat_id=CHAT_ID, text=text, parse_mode=telegram.ParseMode.HTML) |
|
|
|
|
|
def generate_daily_report(): |
|
|
today = datetime.today() |
|
|
start = today.replace(day=1) |
|
|
end = today |
|
|
|
|
|
ruonia_data = get_ruonia(start, end) |
|
|
ruonia_summary, avg_rate, total_vol = summarize_ruonia(ruonia_data) |
|
|
|
|
|
weather = get_weather() |
|
|
|
|
|
prompt = ( |
|
|
f"Ты — эксперт по экономике и геополитике. Вот данные:\n\n" |
|
|
f"{ruonia_summary}\n\n" |
|
|
f"Погода в Москве: {weather}\n\n" |
|
|
f"Сделай краткий вывод на русском: как это связано с ситуацией в стране и мире, кратко, в 3-4 предложениях." |
|
|
) |
|
|
|
|
|
ai_comment = ask_hf(prompt) |
|
|
|
|
|
message = ( |
|
|
"<b>📰 Экономическая сводка за {}</b>\n\n" |
|
|
"{}\n\n" |
|
|
"📊 Средняя ставка RUONIA: {:.2f}%\n" |
|
|
"💰 Общий объём: {:.2f} млн руб.\n" |
|
|
"🌦️ {}\n\n" |
|
|
"🤖 Мнение нейросети:\n{}\n" |
|
|
).format(end.strftime('%d.%m.%Y'), ruonia_summary, avg_rate, total_vol, weather, ai_comment) |
|
|
|
|
|
send_telegram_message(message) |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
generate_daily_report() |
|
|
|
