import os
import logging
import asyncio
import nest_asyncio
import httpx
from telegram import Update, Bot
from telegram.ext import (
Application,
CommandHandler,
MessageHandler,
filters,
CallbackContext
)
# -------------------------
# Configure logging
# -------------------------
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO)
logger = logging.getLogger(__name__)
# -------------------------
# Environment variables
# -------------------------
# Get Telegram bot token from environment variables
TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
if not TOKEN:
raise ValueError("Missing Telegram Bot Token. Please set TELEGRAM_BOT_TOKEN environment variable.")
# # Get the domain for webhook (publicly accessible, e.g., your-space.hf.space)
# WEBHOOK_DOMAIN = os.getenv("WEBHOOK_DOMAIN")
# if not WEBHOOK_DOMAIN:
# raise ValueError("Missing Webhook Domain. Please set WEBHOOK_DOMAIN environment variable.")
# # -------------------------
# # Webhook configuration
# # -------------------------
# # Define a unique webhook path using the bot token
# # WEBHOOK_PATH = f"/{TOKEN}"
# # Construct the full webhook URL (must be HTTPS as required by Telegram)
# WEBHOOK_URL = f"https://{WEBHOOK_DOMAIN}"
# -------------------------
# API URL of the FastAPI server (running on Hugging Face)
# -------------------------
API_URL = "https://demaking-decision-helper-bot.hf.space/generate_response"
# bot = Bot(token=TOKEN)
# -------------------------
# Function to fetch response from FastAPI (unchanged)
# -------------------------
async def fetch_response(user_text: str):
"""
Sends a POST request to the FastAPI API with the user's text and returns the JSON response.
"""
async with httpx.AsyncClient(timeout=45.0) as client:
try:
response = await client.post(API_URL, json={"text": user_text})
response.raise_for_status() # Raise exception for HTTP 4XX/5XX errors
return response.json()
except httpx.HTTPStatusError as e:
logger.error(f"HTTP Error: {e.response.status_code} - {e.response.text}")
return {"response": "Error: API returned an error."}
except httpx.RequestError as e:
logger.error(f"Request Error: {e}")
return {"response": "Error: Could not reach API."}
except Exception as e:
logger.error(f"Unexpected Error: {e}")
return {"response": "Error: Unexpected error occurred."}
# -------------------------
# Command handler for /start
# -------------------------
async def start(update: Update, context: CallbackContext):
""""
Handler for the /start command.
"""
# Respond to the /start command.
await update.message.reply_text("Hello! Tell me your decision-making issue, and I'll try to help.")
logger.info("Start command received.")
# -------------------------
# Message handler for incoming text messages
# -------------------------
async def handle_message(update: Update, context: CallbackContext):
"""
Handler for incoming messages.
Sends the user's message to the API and replies with the response.
"""
user_text = update.message.text
logger.info(f"User message: {user_text}")
# Send the user text to the FastAPI server and get the response.
result = await fetch_response(user_text)
response_text = result.get("response", "Error generating response.")
logger.info(f"API Response: {response_text}")
await update.message.reply_text(response_text)
# def webhook():
# bot.remove_webhook()
# bot.set_webhook(url=WEBHOOL_URL + TOKEN) # הגדר את ה-Webhook
# print("webhook set: ", WEBHOOL_URL + TOKEN)
# return "!", 200
# # -------------------------
# # Set Telegram webhook
# # -------------------------
# async def set_webhook():
# #bot = Bot(token=TOKEN)
# #bot.delete_webhook()
# # This call will set the webhook to the given URL.
# PATH = WEBHOOK_URL + TOKEN
# try:
# await bot.set_webhook(url=PATH)
# logger.info(f"Webhook set successfully to: {WEBHOOK_URL}")
# print("bot webhook success")
# except Exception as e:
# logger.error(f"Failed to set webhook manually. Error: {e}")
# print(f"error setting bot webhook. Error: {e}")
# -------------------------
# Delete Telegram webhook
# -------------------------
# async def delete_webhook():
# try:
# await bot.delete_webhook(drop_pending_updates=True)
# logger.info("Webhook deleted successfully")
# print("deleted webhook successfully")
# except Exception as e:
# logger.error(f"Failed to delete webhook manually. Error: {e}")
# print(f"error deleting bot webhook. Error: {e}")
# -------------------------
# Set Telegram webhook
# -------------------------
# async def set_webhook():
# # bot = Bot(token=TOKEN)
# try:
# await bot.set_webhook(url=WEBHOOK_URL) # This call will set the webhook to the given URL.
# logger.info(f"Webhook set successfully to: {WEBHOOK_URL}")
# print("webhook set succesfully")
# except Exception as e:
# logger.error(f"Failed to set webhook. Error: {e}")
# print(f"error in setting bot webhook. Error: {e}")
# -------------------------
# Main function to run the bot using Webhook mode
# -------------------------
async def main():
"""
Main function to run the Telegram Bot in polling mode.
"""
# await delete_webhook()
# await set_webhook()
# Build the Application with the Telegram Bot Token
application = Application.builder().token(TOKEN).build()
# Add command and message handlers
application.add_handler(CommandHandler("start", start))
application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
# # Log: starting the bot in webhook mode.
# logger.info("Starting bot in webhook mode...")
logger.info("Starting bot in polling mode...")
print("Starting bot in polling mode...")
await application.run_polling()
# Run the application using webhook mode.
# The bot will listen on all interfaces (0.0.0.0) at the specified port.
# await application.run_webhook(
# listen="0.0.0.0", # Listen on all available interfaces
# port=7860, # Port to listen on
# # url_path=TELEGRAM_WEBHOOK, # The webhook path; here, we use the bot token
# webhook_url=WEBHOOK_URL # The webhook URL that Telegram will use to send updates
# )
# -------------------------
# Run the main function
# -------------------------
if __name__ == "__main__":
# Apply nest_asyncio to support nested event loops if required.
nest_asyncio.apply()
try:
print("in try")
loop = asyncio.get_event_loop()
# Schedule the main() task
loop.create_task(main())
loop.run_until_complete(main())
except Exception as e:
logger.error(f"Error in main loop: {e}")
print(f"Error in main loop: {e}")
# try:
# asyncio.run(main())
# except Exception as e:
# logger.error(f"Error in main loop: {e}")
# print(f"Error in main loop: {e}")
# Instead of asyncio.run(), which may try to close an already running loop,
# get the current loop and run main() until complete.
# loop = asyncio.get_event_loop()
# # loop.run_until_complete(main())
# try:
# loop.run_until_complete(main())
# except Exception as e:
# logger.error(f"Error in main loop: {e}")
# print(f"Error in main loop: {e}")
# finally:
# await bot.shutdown()