Upload folder using huggingface_hub

api/canvas.py

@@ -1,76 +1,104 @@
-from fastapi import APIRouter, Depends, HTTPException, status, Response
+from fastapi import APIRouter, Depends, HTTPException, status, Response, BackgroundTasks
 from sqlalchemy.orm import Session
 from typing import List, Optional
 import logging
 from datetime import datetime
 
-from core.database import get_db
+from core.database import get_db, SessionLocal
 from models import db_models
 from models.schemas import CanvasCreateRequest, CanvasEditRequest, CanvasResponse
 from services.canvas_service import canvas_service
 from api.auth import get_current_user
+from api.websocket_routes import manager
 
 router = APIRouter(prefix="/api/canvas", tags=["Canvas - Collaborative Editing"])
 logger = logging.getLogger(__name__)
 
+async def run_canvas_generation(canvas_id: int, request: CanvasCreateRequest, user_id: int):
+    """Background task for canvas summary generation"""
+    db = SessionLocal()
+    connection_id = f"user_{user_id}"
+    try:
+        db_canvas = db.query(db_models.Canvas).filter(db_models.Canvas.id == canvas_id).first()
+        if not db_canvas: return
+
+        # Call AI service
+        content = await canvas_service.generate_canvas_summary(
+            file_key=request.file_key,
+            text_input=request.text_input
+        )
+        
+        if not content:
+            raise Exception("AI failed to generate canvas content")
+
+        db_canvas.text = content
+        db_canvas.status = "completed"
+        db.commit()
+
+        # Notify via WebSocket
+        await manager.send_result(connection_id, {
+            "type": "canvas",
+            "id": db_canvas.id,
+            "status": "completed",
+            "title": db_canvas.title
+        })
+
+    except Exception as e:
+        logger.error(f"Background canvas generation failed: {e}")
+        db_canvas = db.query(db_models.Canvas).filter(db_models.Canvas.id == canvas_id).first()
+        if db_canvas:
+            db_canvas.status = "failed"
+            db_canvas.error_message = str(e)
+            db.commit()
+        await manager.send_error(connection_id, f"Canvas generation failed: {str(e)}")
+    finally:
+        db.close()
+
 @router.post("/create", response_model=CanvasResponse)
 async def create_canvas(
     request: CanvasCreateRequest,
+    background_tasks: BackgroundTasks,
     current_user: db_models.User = Depends(get_current_user),
     db: Session = Depends(get_db)
 ):
     """
-    Create a new Canvas entry. 
-    Pass file_key to extract text and generate an initial Markdown summary.
+    Initiates Canvas creation in the background.
     """
-    try:
-        source_id = None
-        if request.file_key:
-            source = db.query(db_models.Source).filter(
-                db_models.Source.s3_key == request.file_key,
-                db_models.Source.user_id == current_user.id
-            ).first()
-            if not source:
-                raise HTTPException(status_code=404, detail="Source file not found")
-            source_id = source.id
+    source_id = None
+    if request.file_key:
+        source = db.query(db_models.Source).filter(
+            db_models.Source.s3_key == request.file_key,
+            db_models.Source.user_id == current_user.id
+        ).first()
+        if not source:
+            raise HTTPException(status_code=403, detail="Not authorized to access this file")
+        source_id = source.id
+    
+    # Create initial record
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    
+    # Priority: 1. File-based name, 2. User Title (if not default 'string'), 3. Default timestamp
+    if file_base:
+        title = f"Canvas-{file_base}"
+    elif request.title and request.title != "string":
+        title = request.title
+    else:
+        title = f"Canvas {datetime.now().strftime('%Y-%m-%d %H:%M')}"
         
-        # Create initial record
-        title = request.title or f"Canvas {datetime.now().strftime('%Y-%m-%d %H:%M')}"
-        db_canvas = db_models.Canvas(
-            title=title,
-            user_id=current_user.id,
-            source_id=source_id,
-            status="processing"
-        )
-        db.add(db_canvas)
-        db.commit()
-        db.refresh(db_canvas)
-
-        try:
-            # Generate summary
-            content = await canvas_service.generate_canvas_summary(
-                file_key=request.file_key,
-                text_input=request.text_input
-            )
-            
-            db_canvas.text = content
-            db_canvas.status = "completed"
-            db.commit()
-            db.refresh(db_canvas)
-            
-            return db_canvas
-
-        except Exception as e:
-            db_canvas.status = "failed"
-            db_canvas.error_message = str(e)
-            db.commit()
-            raise HTTPException(status_code=500, detail=f"Generation failed: {str(e)}")
+    db_canvas = db_models.Canvas(
+        title=title,
+        user_id=current_user.id,
+        source_id=source_id,
+        status="processing"
+    )
+    db.add(db_canvas)
+    db.commit()
+    db.refresh(db_canvas)
 
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Error creating canvas: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+    # Offload to background task
+    background_tasks.add_task(run_canvas_generation, db_canvas.id, request, current_user.id)
+    
+    return db_canvas
 
 @router.get("/", response_model=List[CanvasResponse])
 async def list_canvases(

api/flashcards.py

@@ -101,7 +101,15 @@ async def generate_flashcards(
         source_id = source.id
 
     # Create initial processing record
-    title = request.topic if request.topic else f"Flashcards {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    
+    # Priority: 1. File-based name, 2. User Topic (if not default 'string'), 3. Default timestamp
+    if file_base:
+        title = f"Flashcard-{file_base}"
+    elif request.topic and request.topic != "string":
+        title = request.topic
+    else:
+        title = f"Flashcards {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_set = db_models.FlashcardSet(
         title=title,
         difficulty=request.difficulty,

api/mindmaps.py

@@ -75,7 +75,15 @@ async def generate_mindmap(
         source_id = source.id
 
     # Create initial processing record
-    title = request.title if request.title else (request.file_key.split('/')[-1] if request.file_key else f"Mind Map {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}")
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    
+    # Priority: 1. File-based name, 2. User Title (if not default 'string'), 3. Default timestamp
+    if file_base:
+        title = f"Mind Map-{file_base}"
+    elif request.title and request.title != "string":
+        title = request.title
+    else:
+        title = f"Mind Map {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_mindmap = db_models.MindMap(
         title=title,
         user_id=current_user.id,

api/podcast.py

@@ -159,8 +159,10 @@ async def generate_podcast(
         source_id = source.id
 
     # 2. Create pending record
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    title = f"Podcast-{file_base}" if file_base else f"Podcast {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_podcast = db_models.Podcast(
-        title=f"Podcast {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}",
+        title=title,
         user_id=current_user.id,
         source_id=source_id,
         status="processing"

api/quizzes.py

@@ -104,7 +104,15 @@ async def generate_quiz(
         source_id = source.id
 
     # Create initial processing record
-    title = request.topic if request.topic else f"Quiz {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    
+    # Priority: 1. File-based name, 2. User Topic (if not default 'string'), 3. Default timestamp
+    if file_base:
+        title = f"Quiz-{file_base}"
+    elif request.topic and request.topic != "string":
+        title = request.topic
+    else:
+        title = f"Quiz {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_set = db_models.QuizSet(
         title=title,
         difficulty=request.difficulty,

api/reports.py

@@ -35,12 +35,12 @@ async def run_report_generation(report_id: int, request: ReportGenerateRequest,
         if not content:
             raise Exception("AI failed to generate report content")
 
-        # Extract title (usually the first line)
-        title = content.split('\n')[0].replace('#', '').strip()
-        if not title or len(title) < 3:
-            title = f"Report {request.format_key}"
-
-        db_report.title = title
+        if not db_report.title or "report-" not in db_report.title:
+            # Extract title (usually the first line)
+            extracted_title = content.split('\n')[0].replace('#', '').strip()
+            if not extracted_title or len(extracted_title) < 3:
+                extracted_title = f"Report {request.format_key}"
+            db_report.title = extracted_title
         db_report.content = content
         db_report.status = "completed"
         db.commit()
@@ -109,7 +109,8 @@ async def generate_report(
         source_id = source.id
 
     # Create initial processing record
-    title = f"Report {request.format_key} {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    title = f"report-{file_base}" if file_base else f"Report {request.format_key} {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_report = db_models.Report(
         title=title,
         format_key=request.format_key,

api/video_generator.py

@@ -43,7 +43,8 @@ async def run_video_generation(summary_id: int, request: VideoSummaryGenerateReq
                 voice_name=request.voice_name
             )
 
-        db_summary.title = result["title"]
+        if not db_summary.title or "Video-" not in db_summary.title:
+            db_summary.title = result["title"]
         db_summary.s3_key = result["s3_key"]
         db_summary.s3_url = result["s3_url"]
         db_summary.status = "completed"
@@ -89,7 +90,8 @@ async def generate_video_summary(
         raise HTTPException(status_code=403, detail="Not authorized to access this file")
 
     # Create initial processing record
-    title = f"Video Summary {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+    file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    title = f"video-{file_base}" if file_base else f"Video Summary {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
     db_summary = db_models.VideoSummary(
         title=title,
         user_id=current_user.id,