Update main.py

main.py

@@ -527,7 +527,7 @@ def generate_image_edit():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-#----------View projedts Endpoint ----------
+#----------View projects and videos Endpoints ----------
 @app.route('/api/view/projects', methods=['GET'])
 def view_projects():
     try:
@@ -555,7 +555,8 @@ def view_projects():
                     "generation_times": story_record.get("generation_times", {}),
                     "created_at": story_record.get("created_at", ""),
                     "input_type": story_record.get("input_type", ""),
-                    "story_type": story_record.get("story_type", "")
+                    "story_type": story_record.get("story_type", ""),
+                    "video_url": story_record.get("video_url", "")  # Include video URL if present
                 }
         
         # Optionally, sort the projects by creation date (newest first)
@@ -567,6 +568,48 @@ def view_projects():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+
+@app.route('/api/view/videos', methods=['GET'])
+def view_videos():
+    """
+    Returns only the stories that have a 'video_url' field,
+    meaning the video has been generated.
+    """
+    try:
+        # --- Authentication ---
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Missing or invalid token'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid or expired token'}), 401
+
+        # --- Query Stories for the Authenticated User ---
+        stories_ref = db.reference('stories')
+        all_stories = stories_ref.get() or {}
+        user_videos = {}
+
+        for story_id, story_record in all_stories.items():
+            # Must belong to current user and have a video_url
+            if story_record.get('uid') == uid and story_record.get('video_url'):
+                user_videos[story_id] = {
+                    "story_id": story_id,
+                    "full_story": story_record.get("full_story", ""),
+                    "sections": story_record.get("sections", []),
+                    "video_url": story_record.get("video_url", ""),
+                    "created_at": story_record.get("created_at", "")
+                }
+
+        # Sort by creation date (newest first), if needed
+        sorted_videos = dict(
+            sorted(user_videos.items(), key=lambda item: item[1]["created_at"], reverse=True)
+        )
+
+        return jsonify({"videos": sorted_videos})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
 # ---------- Credit Request Endpoints ----------
 
 @app.route('/api/user/request-credits', methods=['POST'])