telegram_bot.py

import os
import requests
from flask import Flask, request
from threading import Thread
from config import configure_ai  # Import AI configuration from config.py

# Initialize AI model here
model = configure_ai()

TELEGRAM_TOKEN = os.environ['TELEGRAM_BOT_TOKEN']
BASE_URL = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}"

def send_message(chat_id, text):
    try:
        url = f"{BASE_URL}/sendMessage"
        payload = {
            "chat_id": chat_id,
            "text": text,
            "parse_mode": "Markdown"
        }
        response = requests.post(url, json=payload)
        response.raise_for_status()  # Raise HTTP errors
    except Exception as e:
        logging.error(f"Failed to send message: {str(e)}")

def process_telegram_message(message):
    try:
        chat_id = message['chat']['id']
        doubt = message.get('text', '')
        
        if not doubt:
            send_message(chat_id, "Please provide a valid question.")
            return

        # Generate explanation
        explanation = model.generate_content(
            f"Explain {doubt} in simple terms with examples").text
        
        # Generate practice questions
        questions = model.generate_content(
            f"Generate 3 practice questions about {doubt} with answers").text
        
        response = f"*Explanation*:\n{explanation}\n\n*Practice Questions*:\n{questions}"
        send_message(chat_id, response)

    except Exception as e:
        logging.error(f"Error processing message: {str(e)}")
        send_message(chat_id, "⚠️ Sorry, I encountered an error. Please try again later.")

def setup_telegram_webhook(app):
    @app.route('/telegram-webhook', methods=['POST'])
    def telegram_webhook():
        try:
            update = request.json
            if 'message' in update:
                Thread(target=process_telegram_message, 
                     args=(update['message'],)).start()
            return '', 200
        except Exception as e:
            logging.error(f"Webhook error: {str(e)}")
            return '', 500

    try:
        webhook_url = f"{os.environ['RENDER_EXTERNAL_URL']}/telegram-webhook"
        response = requests.get(f"{BASE_URL}/setWebhook?url={webhook_url}")
        response.raise_for_status()
        logging.info("Webhook setup successful")
    except Exception as e:
        logging.error(f"Webhook setup failed: {str(e)}")