Upload folder using huggingface_hub

api/mindmaps.py

@@ -76,6 +76,7 @@ async def generate_mindmap(
 
     # Create initial processing record
     file_base = request.file_key.split('/')[-1].rsplit('.', 1)[0] if request.file_key else None
+    print(f'file_base: {file_base}')
     
     # Priority: 1. File-based name, 2. User Title (if not default 'string'), 3. Default timestamp
     if file_base:

api/websocket_routes.py

@@ -117,6 +117,12 @@ async def unified_generate_ws(
             
         elif task_type == "mindmap":
             await handle_mindmap_task(connection_id, data, current_user, db)
+
+        elif task_type == "flashcards":
+            await handle_flashcards_task(connection_id, data, current_user, db)
+
+        elif task_type == "quiz":
+            await handle_quiz_task(connection_id, data, current_user, db)
             
         else:
             await manager.send_error(connection_id, f"Unsupported task type: {task_type}")
@@ -146,8 +152,11 @@ async def handle_podcast_task(connection_id: str, data: Dict, current_user: db_m
                 return
             source_id = source.id
         
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("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"
@@ -249,7 +258,15 @@ async def handle_flashcards_task(connection_id: str, data: Dict, current_user: d
             source_id = source.id
         
         # Create initial processing record
-        title = data.get("topic") if data.get("topic") else f"Flashcards {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("file_key") else None
+        
+        if file_base:
+            title = f"Flashcard-{file_base}"
+        elif data.get("topic") and data.get("topic") != "string":
+            title = data.get("topic")
+        else:
+            title = f"Flashcards {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+
         db_set = db_models.FlashcardSet(
             title=title,
             difficulty=data.get("difficulty", "medium"),
@@ -320,7 +337,15 @@ async def handle_quiz_task(connection_id: str, data: Dict, current_user: db_mode
             source_id = source.id
         
         # Create initial processing record
-        title = data.get("topic") if data.get("topic") else f"Quiz {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("file_key") else None
+        
+        if file_base:
+            title = f"Quiz-{file_base}"
+        elif data.get("topic") and data.get("topic") != "string":
+            title = data.get("topic")
+        else:
+            title = f"Quiz {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+
         db_set = db_models.QuizSet(
             title=title,
             difficulty=data.get("difficulty", "medium"),
@@ -381,8 +406,11 @@ async def handle_video_task(connection_id: str, data: Dict, current_user: db_mod
             await manager.send_error(connection_id, "Not authorized to access this file")
             return
 
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("file_key") else None
+        title = f"Video Summary {file_base}" if file_base else f"Video Summary {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+
         db_summary = db_models.VideoSummary(
-            title=f"Video Summary {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}",
+            title=title,
             user_id=current_user.id,
             source_id=source.id,
             status="processing"
@@ -442,8 +470,11 @@ async def handle_report_task(connection_id: str, data: Dict, current_user: db_mo
                 return
             source_id = source.id
 
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("file_key") else None
+        title = f"Report-{file_base}" if file_base else f"Report {data.get('format_key', 'custom')} {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+
         db_report = db_models.Report(
-            title=f"Report {data.get('format_key', 'custom')} {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}",
+            title=title,
             format_key=data.get("format_key", "custom"),
             user_id=current_user.id,
             source_id=source_id,
@@ -466,11 +497,11 @@ 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")
 
-        title = content.split('\n')[0].replace('#', '').strip()
-        if not title or len(title) < 3:
-            title = f"Report {data.get('format_key')}"
-
-        db_report.title = title
+        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:
+                title = f"Report {data.get('format_key')}"
+            db_report.title = title
         db_report.content = content
         db_report.status = "completed"
         db.commit()
@@ -503,8 +534,17 @@ async def handle_mindmap_task(connection_id: str, data: Dict, current_user: db_m
                 return
             source_id = source.id
 
+        file_base = data.get("file_key").split('/')[-1].rsplit('.', 1)[0] if data.get("file_key") else None
+        
+        if file_base:
+            title = f"Mind Map-{file_base}"
+        elif data.get("title") and data.get("title") != "string":
+            title = data.get("title")
+        else:
+            title = f"Mind Map {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}"
+
         db_mindmap = db_models.MindMap(
-            title=data.get("title") if data.get("title") else f"Mind Map {datetime.utcnow().strftime('%Y-%m-%d %H:%M')}",
+            title=title,
             user_id=current_user.id,
             source_id=source_id,
             status="processing"