Merge pull request #430 from CREDO23/feat/chat-pannel

[Feature] Add the chat panel

surfsense_backend/app/tasks/celery_tasks/podcast_tasks.py

@@ -38,7 +38,12 @@ def get_celery_session_maker():
 
 @celery_app.task(name="generate_chat_podcast", bind=True)
 def generate_chat_podcast_task(
-    self, chat_id: int, search_space_id: int, podcast_title: str, user_id: int
+    self,
+    chat_id: int,
+    search_space_id: int,
+    user_id: int,
+    podcast_title: str | None = None,
+    user_prompt: str | None = None,
 ):
     """
     Celery task to generate podcast from chat.
@@ -46,15 +51,18 @@ def generate_chat_podcast_task(
     Args:
         chat_id: ID of the chat to generate podcast from
         search_space_id: ID of the search space
+        user_id: ID of the user,
         podcast_title: Title for the podcast
-        user_id: ID of the user
+        user_prompt: Optional prompt from the user to guide the podcast generation
     """
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
 
     try:
         loop.run_until_complete(
-            _generate_chat_podcast(chat_id, search_space_id, podcast_title, user_id)
+            _generate_chat_podcast(
+                chat_id, search_space_id, user_id, podcast_title, user_prompt
+            )
         )
         loop.run_until_complete(loop.shutdown_asyncgens())
     finally:
@@ -63,13 +71,17 @@ def generate_chat_podcast_task(
 
 
 async def _generate_chat_podcast(
-    chat_id: int, search_space_id: int, podcast_title: str, user_id: int
+    chat_id: int,
+    search_space_id: int,
+    user_id: int,
+    podcast_title: str | None = None,
+    user_prompt: str | None = None,
 ):
     """Generate chat podcast with new session."""
     async with get_celery_session_maker()() as session:
         try:
             await generate_chat_podcast(
-                session, chat_id, search_space_id, podcast_title, user_id
+                session, chat_id, search_space_id, user_id, podcast_title, user_prompt
             )
         except Exception as e:
             logger.error(f"Error generating podcast from chat: {e!s}")

surfsense_backend/app/tasks/podcast_tasks.py

@@ -19,8 +19,9 @@ async def generate_chat_podcast(
     session: AsyncSession,
     chat_id: int,
     search_space_id: int,
-    podcast_title: str,
     user_id: int,
+    podcast_title: str | None = None,
+    user_prompt: str | None = None,
 ):
     task_logger = TaskLoggingService(session, search_space_id)
 
@@ -34,6 +35,7 @@ async def generate_chat_podcast(
             "search_space_id": search_space_id,
             "podcast_title": podcast_title,
             "user_id": str(user_id),
+            "user_prompt": user_prompt,
         },
     )
 
@@ -96,9 +98,10 @@ async def generate_chat_podcast(
 
         config = {
             "configurable": {
-                "podcast_title": "SurfSense",
+                "podcast_title": podcast_title or "SurfSense Podcast",
                 "user_id": str(user_id),
                 "search_space_id": search_space_id,
+                "user_prompt": user_prompt,
             }
         }
         # Initialize state with database session and streaming service
@@ -139,33 +142,49 @@ async def generate_chat_podcast(
             },
         )
 
-        podcast = Podcast(
-            title=f"{podcast_title}",
-            podcast_transcript=serializable_transcript,
-            file_location=result["final_podcast_file_path"],
-            search_space_id=search_space_id,
+        # check if podcast already exists for this chat (re-generation)
+        existing_podcast = await session.execute(
+            select(Podcast).filter(Podcast.chat_id == chat_id)
         )
+        existing_podcast = existing_podcast.scalars().first()
 
-        # Add to session and commit
-        session.add(podcast)
-        await session.commit()
-        await session.refresh(podcast)
+        if existing_podcast:
+            existing_podcast.podcast_transcript = serializable_transcript
+            existing_podcast.file_location = result["final_podcast_file_path"]
+            existing_podcast.chat_state_version = chat.state_version
+            await session.commit()
+            await session.refresh(existing_podcast)
+            return existing_podcast
+        else:
+            podcast = Podcast(
+                title=f"{podcast_title}",
+                podcast_transcript=serializable_transcript,
+                file_location=result["final_podcast_file_path"],
+                search_space_id=search_space_id,
+                chat_state_version=chat.state_version,
+                chat_id=chat.id,
+            )
 
-        # Log success
-        await task_logger.log_task_success(
-            log_entry,
-            f"Successfully generated podcast for chat {chat_id}",
-            {
-                "podcast_id": podcast.id,
-                "podcast_title": podcast_title,
-                "transcript_entries": len(serializable_transcript),
-                "file_location": result.get("final_podcast_file_path"),
-                "processed_messages": processed_messages,
-                "content_length": len(chat_history_str),
-            },
-        )
+            # Add to session and commit
+            session.add(podcast)
+            await session.commit()
+            await session.refresh(podcast)
 
-        return podcast
+            # Log success
+            await task_logger.log_task_success(
+                log_entry,
+                f"Successfully generated podcast for chat {chat_id}",
+                {
+                    "podcast_id": podcast.id,
+                    "podcast_title": podcast_title,
+                    "transcript_entries": len(serializable_transcript),
+                    "file_location": result.get("final_podcast_file_path"),
+                    "processed_messages": processed_messages,
+                    "content_length": len(chat_history_str),
+                },
+            )
+
+            return podcast
 
     except ValueError as ve:
         # ValueError is already logged above for chat not found