Upload folder using huggingface_hub

api/websocket_routes.py

@@ -1,7 +1,6 @@
 import logging
 import asyncio
 from fastapi import APIRouter, WebSocket, WebSocketDisconnect, Depends
-from sqlalchemy import text
 from sqlalchemy.orm import Session
 from datetime import datetime
 from typing import Dict, Any
@@ -220,12 +219,6 @@ async def handle_podcast_task(connection_id: str, data: Dict, current_user: db_m
         
         public_url = s3_service.get_public_url(s3_key)
         
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-            
         db_podcast.s3_key = s3_key
         db_podcast.s3_url = public_url
         db_podcast.script = script
@@ -312,12 +305,6 @@ async def handle_flashcards_task(connection_id: str, data: Dict, current_user: d
             )
             db.add(db_card)
         
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-
         db_set.status = "completed"
         db.commit()
         
@@ -397,12 +384,6 @@ async def handle_quiz_task(connection_id: str, data: Dict, current_user: db_mode
             )
             db.add(db_question)
         
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-
         db_set.status = "completed"
         db.commit()
         await manager.send_result(connection_id, {"id": db_set.id, "title": db_set.title, "status": "completed"})
@@ -454,12 +435,6 @@ async def handle_video_task(connection_id: str, data: Dict, current_user: db_mod
                 voice_name=data.get("voice_name", "Kore")
             )
 
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-
         db_summary.title = result["title"]
         db_summary.s3_key = result["s3_key"]
         db_summary.s3_url = result["s3_url"]
@@ -522,12 +497,6 @@ async def handle_report_task(connection_id: str, data: Dict, current_user: db_mo
         if not content:
             raise Exception("AI failed to generate report content")
 
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-
         if not db_report.title or "Report-" not in db_report.title:
             title = content.split('\n')[0].replace('#', '').strip()
             if not title or len(title) < 3:
@@ -594,12 +563,6 @@ async def handle_mindmap_task(connection_id: str, data: Dict, current_user: db_m
         if not mermaid_code:
             raise Exception("AI failed to generate mind map code")
 
-        # Ensure DB connection is alive before final update
-        try:
-            db.execute(text("SELECT 1"))
-        except Exception:
-            db.rollback()
-
         db_mindmap.mermaid_code = mermaid_code
         db_mindmap.status = "completed"
         db.commit()

core/database.py

@@ -11,13 +11,12 @@ SQLALCHEMY_DATABASE_URL = settings.DATABASE_URL or "sqlite:///./temp.db"
 engine = create_engine(
     SQLALCHEMY_DATABASE_URL,
     pool_pre_ping=True,    # Check connection health before every query
-    pool_recycle=300,      # Refresh connections every 5 minutes (prevents firewall drops)
-    pool_timeout=60,       # Wait up to 60 seconds for a connection
-    pool_size=20,          # Maintain a healthy pool
-    max_overflow=30,       # Allow more overflow if busy
+    pool_recycle=1800,  # 30 minutes (best practice)
+    pool_timeout=30,       # Wait up to 30 seconds for a connection
+    pool_size=15,          # Maintain a slightly larger pool
+    max_overflow=25,       # Allow more overflow if busy
     connect_args={
-        "timeout": 60,     # Command timeout
-        "connect_timeout": 60 # Connection establishment timeout
+        "timeout": 30      # 30 second timeout for queries/connections
     }
 )