reformated the propmt

utils/console.py

@@ -10,28 +10,28 @@ from rich.text import Text
 console = Console()
 
 
-def print_markdown(text):
+def print_markdown(text) -> None:
     """Prints a rich info message. Support Markdown syntax."""
 
     md = Padding(Markdown(text), 2)
     console.print(md)
 
 
-def print_step(text):
+def print_step(text) -> None:
     """Prints a rich info message."""
 
     panel = Panel(Text(text, justify="left"))
     console.print(panel)
 
 
-def print_table(items):
+def print_table(items) -> None:
     """Prints items in a table."""
 
     console.print(Columns([Panel(f"[yellow]{item}", expand=True) for item in items]))
 
 
-def print_substep(text, style=""):
+def print_substep(text, style="") -> None:
-    """Prints a rich info message without the panelling."""
+    """Prints a rich colored info message without the panelling."""
     console.print(text, style=style)
 
 

video_creation/screenshot_downloader.py

@@ -123,21 +123,19 @@ def get_screenshots_of_reddit_posts(reddit_object: dict, screenshot_num: int):
         try:
             page.locator('[data-test-id="post-content"]').screenshot(path=postcontentpath)
         except Exception as e:
-            OKGREEN = '\033[92m'
+            print_substep("Something went wrong!",style="red")
-            WARNING = '\033[93m'
-            ENDC = '\033[0m'
-            print_step(f"{WARNING}Something went wrong!{ENDC}")
             resp = input("Something went wrong with making the screenshots! Do you want to skip the post? (y/n) ")
+
             if resp.casefold().startswith("y"):
                 save_data("", "", "skipped", reddit_id, "")
-                print(f"{OKGREEN}The post is successfully skipped! You can now restart the program and this post will skipped.{ENDC}")
+                print_substep("The post is successfully skipped! You can now restart the program and this post will skipped.","green")
+
             resp = input("Do you want the error traceback for debugging purposes? (y/n)")
-            if resp.casefold().startswith("y"):
+            if not resp.casefold().startswith("y"):
-                print(e)
-                exit()
-            else:
                 exit()
             
+            raise e
+
         if storymode:
             page.locator('[data-click-id="text"]').first.screenshot(
                 path=f"assets/temp/{reddit_id}/png/story_content.png"