Telegram included

app.py

@@ -5,6 +5,7 @@ import torch
 import logging
 import re
 import json
+import asyncio
 from typing import Dict, Optional, List, Union
 from fastapi import FastAPI, Request, BackgroundTasks, HTTPException, Depends
 from fastapi.responses import JSONResponse
@@ -20,6 +21,8 @@ API_KEY = os.getenv("API_KEY", "default-key-123")
 MAX_TOKENS = 450
 DEVICE = "cpu"
 PORT = int(os.getenv("PORT", 7860))
+TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", "")
+TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID", "")
 
 # Setup logging
 logging.basicConfig(
@@ -85,6 +88,34 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
+async def send_telegram_message(message: str):
+    """Send message to Telegram"""
+    if not TELEGRAM_BOT_TOKEN or not TELEGRAM_CHAT_ID:
+        logger.warning("Telegram credentials not configured")
+        return False
+        
+    try:
+        url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
+        payload = {
+            "chat_id": TELEGRAM_CHAT_ID,
+            "text": message,
+            "parse_mode": "HTML",
+            "disable_web_page_preview": True
+        }
+        
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            response = await client.post(url, json=payload)
+            if response.status_code == 200:
+                logger.info("📨 Telegram notification sent")
+                return True
+            else:
+                logger.error(f"❌ Telegram API error: {response.status_code} - {response.text}")
+                return False
+                
+    except Exception as e:
+        logger.error(f"❌ Telegram notification failed: {str(e)}")
+        return False
+
 def extract_topics(topic_input: Union[str, List[str]]) -> List[str]:
     """Extract and validate topics from input"""
     if isinstance(topic_input, str):
@@ -273,6 +304,10 @@ async def process_job(job_id: str, topics_input: Union[str, List[str]], callback
             
         logger.info(f"🎯 Processing {len(topics)} topics: {topics}")
         
+        # Send start notification to Telegram
+        start_message = f"🚀 <b>Script Generation Started</b>\n\nTopics: {', '.join(topics[:3])}{'...' if len(topics) > 3 else ''}"
+        asyncio.create_task(send_telegram_message(start_message))
+        
         # Step 1: Generate a viral topic from the trends
         generated_topic = generate_topic_from_trends(topics)
         
@@ -291,6 +326,19 @@ async def process_job(job_id: str, topics_input: Union[str, List[str]], callback
         
         logger.info(f"✅ Completed job {job_id}")
         
+        # Send completion notification to Telegram
+        completion_message = f"""
+✅ <b>Script Generation Complete!</b>
+
+🏷️ <b>Generated Topic:</b> {generated_topic}
+
+📝 <b>Script Length:</b> {len(script)} characters
+🕒 <b>Estimated Video Duration:</b> 60 seconds
+
+🔗 <b>View in n8n:</b> Check your workflow for the complete script
+        """
+        asyncio.create_task(send_telegram_message(completion_message))
+        
         # Send webhook callback if URL provided
         if callback_url:
             try:
@@ -320,6 +368,10 @@ async def process_job(job_id: str, topics_input: Union[str, List[str]], callback
         error_msg = f"Job failed: {str(e)}"
         logger.error(f"❌ Job {job_id} failed: {error_msg}")
         
+        # Send error notification to Telegram
+        error_message = f"❌ <b>Script Generation Failed</b>\n\nError: {str(e)}"
+        asyncio.create_task(send_telegram_message(error_message))
+        
         # Store failure information
         jobs[job_id] = {
             "status": "failed",
@@ -454,26 +506,29 @@ async def test_generation(auth: bool = Depends(verify_api_key)):
         logger.error(f"❌ Test generation failed: {str(e)}")
         return {"status": "error", "error": str(e)}
 
+@app.get("/test/telegram")
+async def test_telegram(auth: bool = Depends(verify_api_key)):
+    """Test Telegram notification"""
+    test_message = "🔔 <b>Test Notification</b>\n\nThis is a test message from your Script Generator API"
+    success = await send_telegram_message(test_message)
+    
+    return {
+        "status": "success" if success else "error",
+        "message": "Telegram test sent" if success else "Telegram test failed",
+        "bot_configured": bool(TELEGRAM_BOT_TOKEN and TELEGRAM_CHAT_ID)
+    }
+
 @app.get("/")
 async def root():
     """Root endpoint"""
     return {
-        "message": "Enhanced Video Script Generator API",
-        "version": "2.0",
-        "features": "Generates viral topics from trends, then creates scripts",
+        "message": "Enhanced Video Script Generator API with Telegram Notifications",
+        "version": "2.1",
+        "features": "Generates viral topics from trends, creates scripts, and sends Telegram notifications",
         "endpoints": {
             "submit_job": "POST /api/submit (with 'topics' array)",
             "check_status": "GET /api/status/{job_id}",
             "health": "GET /health",
-            "test": "GET /test/generation"
-        },
-        "status": "operational"
-    }
-
-if __name__ == "__main__":
-    uvicorn.run(
-        app,
-        host="0.0.0.0",
-        port=PORT,
-        log_level="info"
-    )
+            "test_generation": "GET /test/generation",
+            "test_telegram": "GET /test/telegram"
+        },