blyablya

bot.py

@@ -1,17 +1,24 @@
-import os
-import telebot
-from huggingface_hub import InferenceClient
+from huggingface_hub import HfApi, InferenceApi
 
-# Получаем токены из окружения
-hf_token = os.getenv("HUGGINGFACE_TOKEN")
-tg_token = os.getenv("TG_TOKEN")
+# Получаем токен из secrets
+token = os.getenv("HUGGINGFACE_TOKEN")
 
-bot = telebot.TeleBot(tg_token)
-client = InferenceClient("sambanovasystems/SambaLingo-Russian-Chat", token=hf_token)
+# Проверяем наличие токена
+if not token:
+    raise ValueError("HUGGINGFACE_TOKEN is not set in secrets.")
 
-@bot.message_handler(func=lambda message: True)
-def echo_all(message):
-    response = client.chat_completion(messages=[{"role": "user", "content": message.text}])
-    bot.reply_to(message, response['choices'][0]['message']['content'])
+# Инициализируем Inference API
+api = InferenceApi(repo_id="sambanovasystems/SambaLingo-Russian-Chat", token=token)
 
-bot.polling()
+def respond(user_message):
+    try:
+        response = api(inputs=user_message)
+        return response
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return "Произошла ошибка при обращении к модели."
+
+# Пример использования
+user_message = "Завершите параграф: Родион Раскольников - главный персонаж романа 'Преступление и наказание' Федора Достоевского."
+result = respond(user_message)
+print(result)