added video id to search results

NotJoeMartinez · Sep 6, 2024 · 2df534d · 2df534d
1 parent cafd46f
commit 2df534d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/yt_fts/search.py b/yt_fts/search.py
@@ -159,6 +159,7 @@ def print_fts_res(self):
             metadata = quote["metadata"]
             video_name = metadata["video_title"]
             video_date = metadata["video_date"]
+            video_id = quote["video_id"]
             quote_data = {
                 "quote": quote["subs"],
                 "time_stamp": quote["time_stamp"],
@@ -167,8 +168,8 @@ def print_fts_res(self):
             if channel_name not in fts_dict:
                 fts_dict[channel_name] = {}
             if (video_name, video_date) not in fts_dict[channel_name]:
-                fts_dict[channel_name][(video_name, video_date)] = []
-            fts_dict[channel_name][(video_name, video_date)].append(quote_data)
+                fts_dict[channel_name][(video_name, video_date, video_id)] = []
+            fts_dict[channel_name][(video_name, video_date, video_id)].append(quote_data)
 
         # Sort the list by the total number of quotes in each channel
         channel_list = list(fts_dict.items())
@@ -182,8 +183,8 @@ def print_fts_res(self):
             video_list = list(videos.items())
             video_list.sort(key=lambda x: len(x[1]))
 
-            for (video_name, video_date), quotes in video_list:
-                console.print(f"    [bold][blue]{video_name}[/blue][/bold] ({video_date})")
+            for (video_name, video_date, video_id), quotes in video_list:
+                console.print(f"{video_id} ({video_date}): \"[bold][blue]{video_name}[/blue][/bold]\"")
                 console.print("")
 
                 # Sort the quotes by timestamp

diff --git a/yt_fts/summarize.py b/yt_fts/summarize.py
@@ -5,6 +5,7 @@
 
 import yt_dlp
 from rich.console import Console
+from rich.markdown import Markdown
 from urllib.parse import urlparse, parse_qs
 
 from .config import get_db_path
@@ -60,7 +61,8 @@ def summarize_video(self):
         ]
 
         summary_text = self.get_completion(messages)
-        console.print(summary_text)
+        md = Markdown(summary_text)
+        self.console.print(md)
 
 
     def get_completion(self, messages: list) -> str: