First

utils/story_management.py

@@ -23,6 +23,13 @@ def log_execution(func):
         end_str = datetime.fromtimestamp(end_time).strftime('%Y-%m-%d %H:%M:%S')
         duration = end_time - start_time
 
+        # Write to file (works in Colab)
+        with open('content/logs.txt', 'a') as f:
+            f.write(f"{func.__name__}, start: {start_str}, end: {end_str}, duration: {duration:.4f}s\n")
+
+        # Also print to see output immediately
+        print(f"{func.__name__}, start: {start_str}, end: {end_str}, duration: {duration:.4f}s")
+
         return result
 
     return wrapper
@@ -224,7 +231,7 @@ def generate_direct_comic(
         error_details = traceback.format_exc()
         print(f"❌ Error generating story scenes: {e}")
         print(f"Full error details: {error_details}")
-        pass
+
         return None, "", f"❌ Error generating story scenes: {str(e)}", ""
 
 
@@ -443,29 +450,29 @@ def generate_image_narration(image_path, narration_length=3, age_group: str | No
     return ""
 
 
-# @log_execution
-# def load_narration_from_file(comic_image_path):
-#     """
-#     Load narration from the saved narration.txt file for a given comic image.
-#
-#     Args:
-#         comic_image_path: Path to the comic image
-#
-#     Returns:
-#         str: Loaded narration text or empty string if not found
-#     """
-#     try:
-#         if not comic_image_path:
-#             return ""
-#
-#         image_dir = os.path.dirname(comic_image_path)
-#         narration_path = os.path.join(image_dir, "narration.txt")
-#
-#         if os.path.exists(narration_path):
-#             with open(narration_path, "r", encoding="utf-8") as f:
-#                 narration = f.read().strip()
-#             return narration
-#     except Exception as e:
-#         print(f"⚠️ Could not load narration: {e}")
-#
-#     return ""
+@log_execution
+def load_narration_from_file(comic_image_path):
+    """
+    Load narration from the saved narration.txt file for a given comic image.
+
+    Args:
+        comic_image_path: Path to the comic image
+
+    Returns:
+        str: Loaded narration text or empty string if not found
+    """
+    try:
+        if not comic_image_path:
+            return ""
+
+        image_dir = os.path.dirname(comic_image_path)
+        narration_path = os.path.join(image_dir, "narration.txt")
+
+        if os.path.exists(narration_path):
+            with open(narration_path, "r", encoding="utf-8") as f:
+                narration = f.read().strip()
+            return narration
+    except Exception as e:
+        print(f"⚠️ Could not load narration: {e}")
+
+    return ""