Update main.py

main.py

@@ -494,6 +494,45 @@ def generate_image_edit():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+#----------View projedts Endpoint ----------
+@app.route('/api/view/projects', methods=['GET'])
+def view_projects():
+    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_stories = {}
+        
+        for story_id, story_record in all_stories.items():
+            # Only return projects that belong to the current user
+            if story_record.get('uid') == uid:
+                user_stories[story_id] = {
+                    "story_id": story_id,
+                    "full_story": story_record.get("full_story", ""),
+                    "sections": story_record.get("sections", []),
+                    "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", "")
+                }
+        
+        # Optionally, sort the projects by creation date (newest first)
+        sorted_stories = dict(
+            sorted(user_stories.items(), key=lambda item: item[1]["created_at"], reverse=True)
+        )
+        
+        return jsonify({"projects": sorted_stories})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
 
 # ---------- Credit Request Endpoints ----------