fix dockerfile and bot

Dockerfile

@@ -37,6 +37,9 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 # Copy source code with proper ownership
 COPY --chown=user src/ $HOME/app/src/
 
+# Copy the main application file
+COPY --chown=user main.py $HOME/app/
+
 # Copy any additional files you need
 COPY --chown=user *.py $HOME/app/
 COPY --chown=user .env* $HOME/app/
@@ -55,6 +58,6 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
 
 # Run the application when the container starts HF
 #CMD ["uvicorn", "src.telegram_bot:app", "--host", "0.0.0.0", "--port", "7860"]
-CMD ["python", "telegram_bot.py"]
+CMD ["python", "main.py"]
 # Uncomment the line below to run the application locally
 #CMD ["python", "-m", "telegram_bot"]

main.py

@@ -0,0 +1,5 @@
+from src.telegram_bot import main as telegram_bot
+
+
+if __name__ == "__main__":
+    telegram_bot()

src/telegram_bot.py

@@ -99,13 +99,13 @@ app = FastAPI(title="Financial News Bot", description="Telegram bot for financia
 
 
 @app.on_event("startup")
-async def startup_event():
+async def startup_event() -> None:
     """Initialize Telegram bot on FastAPI startup"""
     await init_telegram_app()
 
 
 @app.on_event("shutdown")
-async def shutdown_event():
+async def shutdown_event() -> None:
     """Cleanup on shutdown"""
     global application
     if application:
@@ -113,7 +113,7 @@ async def shutdown_event():
 
 
 @app.post(f"/{BOT_TOKEN}")
-async def webhook(request: Request):
+async def webhook(request: Request) -> dict[str, Any]:
     """Handle incoming webhook from Telegram"""
     try:
         # Get the update from Telegram
@@ -130,7 +130,7 @@ async def webhook(request: Request):
 
 
 @app.get("/")
-async def root():
+async def root() -> dict[str, Any]:
     """Health check endpoint"""
     return {
         "status": "Financial News Bot is running!",
@@ -142,7 +142,7 @@ async def root():
 
 
 @app.get("/set_webhook")
-async def set_webhook():
+async def set_webhook() -> dict[str, Any]:
     """Manually set the webhook (call this once after deployment)"""
     try:
         webhook_url = f"https://{SPACE_ID}.hf.space/{BOT_TOKEN}"
@@ -170,7 +170,7 @@ async def set_webhook():
 
 
 @app.get("/webhook_info")
-async def webhook_info():
+async def webhook_info() -> dict[str, Any]:
     """Get current webhook information"""
     try:
         info_url = f"https://api.telegram.org/bot{BOT_TOKEN}/getWebhookInfo"
@@ -186,12 +186,12 @@ async def webhook_info():
 
 
 @app.get("/health")
-async def health():
+async def health() -> dict[str, Any]:
     """Additional health check"""
     return {"status": "healthy", "bot_token_set": bool(BOT_TOKEN)}
 
 
-if __name__ == "__main__":
+def main() -> None:
     logger.info(f"Starting Financial News Bot on port {PORT}")
     logger.info(f"Bot token: {BOT_TOKEN[:10]}..." if BOT_TOKEN else "No token set")
     logger.info(f"Space ID: {SPACE_ID}")