Update backend.py

backend.py

@@ -1,140 +1,140 @@
-import os
-import logging
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-from pydantic import BaseModel
-from typing import List, Optional
-from dotenv import load_dotenv
-from utils.youtube_transcript import get_youtube_transcript
-from utils.script_generator import generate_script
-from deep_translator import GoogleTranslator
-from langdetect import detect
-
-# Set up logging
-logging.basicConfig(level=logging.INFO, 
-                   format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-logger = logging.getLogger(__name__)
-
-# Load environment variables
-load_dotenv()
-
-# Initialize FastAPI app
-app = FastAPI(title="Script Generator API")
-
-# Add CORS middleware to allow frontend to access
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  # For production, replace with specific origins
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Add error handling middleware
-@app.middleware("http")
-async def log_exceptions(request: Request, call_next):
-    try:
-        return await call_next(request)
-    except Exception as e:
-        logger.error(f"Unhandled exception: {str(e)}")
-        return JSONResponse(
-            status_code=500,
-            content={"detail": f"Internal Server Error: {str(e)}"}
-        )
-
-# Pydantic models
-class ScriptRequest(BaseModel):
-    video_url: Optional[str] = None
-    posts: Optional[List[str]] = None
-    topic: str
-
-class ScriptResponse(BaseModel):
-    script: str
-
-def translate_text(text, target_lang='en'):
-    """Translate text to the target language if needed"""
-    try:
-        if not text or text.strip() == "":
-            return ""
-            
-        source_lang = detect(text)
-        if source_lang == target_lang:
-            return text
-        translator = GoogleTranslator(source=source_lang, target=target_lang)
-        return translator.translate(text)
-    except Exception as e:
-        logger.error(f"Translation error: {str(e)}")
-        return f"Translation error: {str(e)}"
-
-@app.post("/generate-script", response_model=ScriptResponse)
-async def create_script(request: ScriptRequest):
-    """Generate a script based on YouTube video or example posts"""
-    logger.info(f"Received script generation request: {request}")
-    examples = []
-    
-    # Validate topic
-    if not request.topic or request.topic.strip() == "":
-        raise HTTPException(status_code=400, detail="Topic cannot be empty")
-    
-    if request.video_url:
-        logger.info(f"Fetching transcript for video: {request.video_url}")
-        transcript = get_youtube_transcript(request.video_url)
-        if isinstance(transcript, str) and transcript.startswith("Error:"):
-            logger.error(f"Transcript error: {transcript}")
-            raise HTTPException(status_code=400, detail=transcript)
-        
-        # Translate transcript to English if needed
-        logger.info("Translating transcript if needed")
-        transcript = translate_text(transcript)
-        examples = [transcript]
-    elif request.posts and len(request.posts) >= 3:
-        logger.info(f"Processing {len(request.posts)} example posts")
-        # Translate posts to English if needed
-        valid_posts = [post for post in request.posts if post and post.strip()]
-        if len(valid_posts) < 3:
-            raise HTTPException(status_code=400, detail="Please provide at least three non-empty example posts")
-            
-        # Translate posts to English if needed
-        translated_posts = [translate_text(post) for post in valid_posts[:3]]
-        examples = translated_posts
-    else:
-        logger.error("No valid input provided")
-        raise HTTPException(
-            status_code=400, 
-            detail="Please provide either a YouTube video URL or at least three example posts"
-        )
-    
-    try:
-        # Generate script
-        logger.info(f"Generating script with {len(examples)} examples and topic '{request.topic}'")
-        script = generate_script(None, examples, request.topic)
-        return ScriptResponse(script=script)
-    except Exception as e:
-        logger.error(f"Error generating script: {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Error generating script: {str(e)}")
-
-@app.get("/health")
-async def health_check():
-    """Health check endpoint"""
-    # Check if GROQ_API_KEY is set
-    groq_api_key = os.getenv("GROQ_API_KEY")
-    if not groq_api_key:
-        logger.warning("GROQ_API_KEY environment variable is not set")
-        return {"status": "warning", "message": "GROQ_API_KEY not set"}
-    
-    return {"status": "healthy"}
-
-# Run the server
-if __name__ == "__main__":
-    import uvicorn
-    # Log GROQ_API_KEY status (don't log the actual key)
-    groq_api_key = os.getenv("GROQ_API_KEY")
-    if groq_api_key:
-        logger.info("GROQ_API_KEY is set")
-    else:
-        logger.warning("GROQ_API_KEY environment variable is not set!")
-    
-    # Run on port 8000
-    logger.info("Starting FastAPI server on port 8000")
-    uvicorn.run("backend:app", host="0.0.0.0", port=8000, reload=True) 
+import os
+import logging
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from typing import List, Optional
+from dotenv import load_dotenv
+from utils.youtube_transcript import get_youtube_transcript
+from utils.script_generator import generate_script
+from deep_translator import GoogleTranslator
+from langdetect import detect
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, 
+                   format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# Load environment variables
+load_dotenv()
+
+# Initialize FastAPI app
+app = FastAPI(title="Script Generator API")
+
+# Add CORS middleware to allow frontend to access
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # For production, replace with specific origins
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Add error handling middleware
+@app.middleware("http")
+async def log_exceptions(request: Request, call_next):
+    try:
+        return await call_next(request)
+    except Exception as e:
+        logger.error(f"Unhandled exception: {str(e)}")
+        return JSONResponse(
+            status_code=500,
+            content={"detail": f"Internal Server Error: {str(e)}"}
+        )
+
+# Pydantic models
+class ScriptRequest(BaseModel):
+    video_url: Optional[str] = None
+    posts: Optional[List[str]] = None
+    topic: str
+
+class ScriptResponse(BaseModel):
+    script: str
+
+def translate_text(text, target_lang='en'):
+    """Translate text to the target language if needed"""
+    try:
+        if not text or text.strip() == "":
+            return ""
+            
+        source_lang = detect(text)
+        if source_lang == target_lang:
+            return text
+        translator = GoogleTranslator(source=source_lang, target=target_lang)
+        return translator.translate(text)
+    except Exception as e:
+        logger.error(f"Translation error: {str(e)}")
+        return f"Translation error: {str(e)}"
+
+@app.post("/generate-script", response_model=ScriptResponse)
+async def create_script(request: ScriptRequest):
+    """Generate a script based on YouTube video or example posts"""
+    logger.info(f"Received script generation request: {request}")
+    examples = []
+    
+    # Validate topic
+    if not request.topic or request.topic.strip() == "":
+        raise HTTPException(status_code=400, detail="Topic cannot be empty")
+    
+    if request.video_url:
+        logger.info(f"Fetching transcript for video: {request.video_url}")
+        transcript = get_youtube_transcript(request.video_url)
+        if isinstance(transcript, str) and transcript.startswith("Error:"):
+            logger.error(f"Transcript error: {transcript}")
+            raise HTTPException(status_code=400, detail=transcript)
+        
+        # Translate transcript to English if needed
+        logger.info("Translating transcript if needed")
+        transcript = translate_text(transcript)
+        examples = [transcript]
+    elif request.posts and len(request.posts) >= 3:
+        logger.info(f"Processing {len(request.posts)} example posts")
+        # Translate posts to English if needed
+        valid_posts = [post for post in request.posts if post and post.strip()]
+        if len(valid_posts) < 3:
+            raise HTTPException(status_code=400, detail="Please provide at least three non-empty example posts")
+            
+        # Translate posts to English if needed
+        translated_posts = [translate_text(post) for post in valid_posts[:3]]
+        examples = translated_posts
+    else:
+        logger.error("No valid input provided")
+        raise HTTPException(
+            status_code=400, 
+            detail="Please provide either a YouTube video URL or at least three example posts"
+        )
+    
+    try:
+        # Generate script
+        logger.info(f"Generating script with {len(examples)} examples and topic '{request.topic}'")
+        script = generate_script(None, examples, request.topic)
+        return ScriptResponse(script=script)
+    except Exception as e:
+        logger.error(f"Error generating script: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error generating script: {str(e)}")
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint"""
+    # Check if GROQ_API_KEY is set
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    if not groq_api_key:
+        logger.warning("GROQ_API_KEY environment variable is not set")
+        return {"status": "warning", "message": "GROQ_API_KEY not set"}
+    
+    return {"status": "healthy"}
+
+# Run the server
+if __name__ == "__main__":
+    import uvicorn
+    # Log GROQ_API_KEY status (don't log the actual key)
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    if groq_api_key:
+        logger.info("GROQ_API_KEY is set")
+    else:
+        logger.warning("GROQ_API_KEY environment variable is not set!")
+    
+    # Run on port 8000
+    logger.info("Starting FastAPI server on port 7860")
+    uvicorn.run("backend:app", host="0.0.0.0", port=7860, reload=True)