Create bot.py

bot.py

@@ -0,0 +1,71 @@
+import os
+import logging
+import asyncio
+import nest_asyncio
+from telegram import Update
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes
+
+from utils import clean_text
+from app import generate_response  # Import the response generation function from app.py
+
+
+# Configure logging
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+
+# Get Telegram Bot token from environment variable
+TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+if not TOKEN:
+    raise ValueError("No TELEGRAM_BOT_TOKEN found in environment variables!")
+
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """
+    Handler for the /start command.
+    Replies with a welcome message.
+    """
+    if not update.message or not update.message.text:
+        return
+    await update.message.reply_text("Hello! Tell me your decision-making issue, and I'll try to help.")
+    logger.info("Start command received.")
+
+
+async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """
+    Handler for incoming messages.
+    Cleans the text, generates a response, and replies to the user.
+    """
+    try:
+        if not update.message or not update.message.text:
+            return
+        user_text = clean_text(update.message.text)
+        logger.info(f"User message: {user_text}")
+
+        # Generate response using the function from app.py
+        response = generate_response(user_text)
+        await update.message.reply_text(response)
+    except Exception as e:
+        logger.error(f"Error processing message: {e}")
+
+
+def main():
+    """
+    Main function to start the Telegram bot in polling mode.
+    """
+    application = Application.builder().token(TOKEN).build()
+    application.add_handler(CommandHandler("start", start))
+    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+
+    # Start the bot using polling mode.
+    application.run_polling(allowed_updates=Update.ALL_TYPES)
+
+
+if __name__ == '__main__':
+    # Apply nest_asyncio to support nested event loops in HF Spaces
+    nest_asyncio.apply()
+    # For HF Spaces, using run_polling() is simpler and less dependent on DNS/HTTPS issues.
+    main()