fix tg bot

requirements.txt

@@ -112,4 +112,6 @@ isort>=5.10.0
 
 finnhub-python>=2.4.0
 python-telegram-bot>=20.0
-httpx==0.25.0
+httpx==0.25.0
+fastapi==0.104.1
+uvicorn==0.24.0

src/telegram_bot.py

@@ -7,6 +7,9 @@ from telegram import Update
 from telegram.ext import Application, ApplicationBuilder, CommandHandler, ContextTypes
 from dotenv import load_dotenv
 from src.financial_news_requester import fetch_comp_financial_news
+from fastapi import FastAPI, Request
+import uvicorn
+import asyncio
 
 
 logging.basicConfig(level=logging.INFO)
@@ -20,6 +23,7 @@ if not TELEGRAM_TOKEN:
     logger.error("TELEGRAM_TOKEN not found! Please add it in Space Settings > Repository secrets")
     raise ValueError("TELEGRAM_TOKEN not found in environment variables")
 SPACE_ID = os.environ.get('SPACE_ID', 'ResearchEngineering/news_sentiment_analyzer')
+PORT = int(os.environ.get('PORT', 7860))
 
 
 def format_news_for_telegram(news_json: list[dict[str, Any]]) -> str:
@@ -37,6 +41,10 @@ def format_news_for_telegram(news_json: list[dict[str, Any]]) -> str:
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
     await update.message.reply_text("Hello! I'm your Financial Bot.")
 
+async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    """Handle /help command"""
+    await update.message.reply_text("🤖 Available commands:\n/start - Start the bot\n/help - Show this help")
+
 async def run_crew(update: Update, context: ContextTypes.DEFAULT_TYPE):
     await update.message.reply_text("Running ...")
     try:
@@ -47,6 +55,58 @@ async def run_crew(update: Update, context: ContextTypes.DEFAULT_TYPE):
         await update.message.reply_text(f"Error: {e}")
 
 
+app = FastAPI()
+
+
+@app.post(f"/{TELEGRAM_TOKEN}")
+async def webhook(request: Request):
+    """Handle incoming webhook from Telegram"""
+    try:
+        # Get the update from Telegram
+        json_data = await request.json()
+        update = Update.de_json(json_data, application.bot)
+
+        # Process the update
+        await application.process_update(update)
+
+        return {"status": "ok"}
+    except Exception as e:
+        logger.error(f"Error processing update: {e}")
+        return {"status": "error", "message": str(e)}
+
+
+@app.get("/")
+async def root():
+    """Health check endpoint"""
+    return {"status": "Bot is running!", "webhook_url": f"https://{SPACE_ID}.hf.space/{TELEGRAM_TOKEN}"}
+
+
+@app.get("/set_webhook")
+async def set_webhook():
+    """Manually set the webhook (call this once after deployment)"""
+    try:
+        webhook_url = f"https://{SPACE_ID}.hf.space/{TELEGRAM_TOKEN}"
+
+        # Use a simple HTTP request to set webhook instead of bot.set_webhook()
+        import httpx
+
+        set_webhook_url = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}/setWebhook"
+
+        async with httpx.AsyncClient() as client:
+            response = await client.post(set_webhook_url, json={"url": webhook_url})
+            result = response.json()
+
+        if result.get("ok"):
+            return {"status": "success", "webhook_url": webhook_url, "result": result}
+        else:
+            return {"status": "error", "result": result}
+
+    except Exception as e:
+        logger.error(f"Error setting webhook: {e}")
+        return {"status": "error", "message": str(e)}
+
+
+
 def func():
     logging.basicConfig(level=logging.INFO)
     app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
@@ -55,7 +115,7 @@ def func():
     app.run_polling()
 
 
-if __name__ == "__main__":
+def test_func():
     application = Application.builder().token(TELEGRAM_TOKEN).build()
     application.add_handler(CommandHandler("start", start))
     PORT = int(os.environ.get('PORT', 8000))
@@ -67,3 +127,16 @@ if __name__ == "__main__":
         url_path=TELEGRAM_TOKEN,
         webhook_url=f"https://{os.environ.get('SPACE_ID', SPACE_ID)}.hf.space/{TELEGRAM_TOKEN}"
     )
+
+
+if __name__ == "__main__":
+    # Create telegram application WITHOUT initializing
+    application = Application.builder().token(TELEGRAM_TOKEN).build()
+    application.add_handler(CommandHandler("start", start))
+    application.add_handler(CommandHandler("help", help_command))
+
+    logger.info(f"Starting bot on port {PORT}")
+    logger.info(f"Webhook URL will be: https://{SPACE_ID}.hf.space/{TELEGRAM_TOKEN}")
+    logger.info("After deployment, visit /set_webhook to configure the webhook")
+
+    uvicorn.run(app, host="0.0.0.0", port=PORT)