Update main.py

main.py

@@ -109,7 +109,7 @@ def signup():
         user_ref = db.reference(f'users/{user.uid}')
         user_data = {
             'email': email,
-            'credits': 3,
+            'credits': 30,
             'is_admin': False,
             'created_at': datetime.utcnow().isoformat()
         }
@@ -155,9 +155,182 @@ def get_user_profile():
     except Exception as e:
         print(f"Error fetching user profile: {str(e)}")
         return jsonify({'error': str(e)}), 500
+
+
+
+# -----------------------
+# Content
+# -----------------------
+
 # ---------- Video Generation Endpoint ----------
 
 @app.route('/api/video/generate', methods=['POST'])
+def generate_video():
+    try:
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Authorization header missing or malformed'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid token'}), 401
+
+        # Get user data and check credits
+        user_ref = db.reference(f'users/{uid}')
+        user_data = user_ref.get()
+        if not user_data or user_data.get('credits', 0) < 1:
+            return jsonify({'error': 'Insufficient credits'}), 400
+
+        # Deduct one credit
+        new_credits = user_data.get('credits', 0) - 5
+        user_ref.update({'credits': new_credits})
+
+        # Get prompt and generate video (simulation using Gemini)
+        req_data = request.get_json()
+        prompt = req_data.get('prompt', '')
+        if not prompt:
+            return jsonify({'error': 'Prompt is required'}), 400
+
+        generated_video = ""
+
+        return jsonify({
+            'success': True,
+            'video': generated_video,
+            'remaining_credits': new_credits
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+# ---------- Story Generation Endpoint ----------
+
+@app.route('/api/story/generate', methods=['POST'])
+def generate_story():
+    try:
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Authorization header missing or malformed'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid token'}), 401
+
+        # Get user data and check credits
+        user_ref = db.reference(f'users/{uid}')
+        user_data = user_ref.get()
+        if not user_data or user_data.get('credits', 0) < 1:
+            return jsonify({'error': 'Insufficient credits'}), 400
+
+        # Deduct one credit
+        new_credits = user_data.get('credits', 0) - 5
+        user_ref.update({'credits': new_credits})
+
+        # Get prompt and generate video (simulation using Gemini)
+        req_data = request.get_json()
+        prompt = req_data.get('prompt', '')
+        if not prompt:
+            return jsonify({'error': 'Prompt is required'}), 400
+
+        model = configure_gemini()
+        # Call to Gemini to generate video content (this is a simulation)
+        response = "function(s) that does something"
+        # For demonstration, we assume the response contains a 'result' field
+        generated_story = response.result if hasattr(response, 'result') else "story generated based on parameters."
+
+        return jsonify({
+            'success': True,
+            'story': generated_story,
+            'remaining_credits': new_credits
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+# ---------- Audio Generation Endpoint ----------
+@app.route('/api/audio/generate', methods=['POST'])
+def generate_video():
+    try:
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Authorization header missing or malformed'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid token'}), 401
+
+        # Get user data and check credits
+        user_ref = db.reference(f'users/{uid}')
+        user_data = user_ref.get()
+        if not user_data or user_data.get('credits', 0) < 1:
+            return jsonify({'error': 'Insufficient credits'}), 400
+
+        # Deduct one credit
+        new_credits = user_data.get('credits', 0) - 1
+        user_ref.update({'credits': new_credits})
+
+        # Get prompt and generate video (simulation using Gemini)
+        req_data = request.get_json()
+        prompt = req_data.get('prompt', '')
+        if not prompt:
+            return jsonify({'error': 'Prompt is required'}), 400
+
+        model = configure_gemini()
+        # Call to Gemini to generate video content (this is a simulation)
+        response = model.generate(prompt=prompt)
+        # For demonstration, we assume the response contains a 'result' field
+        generated_video = response.result if hasattr(response, 'result') else "Video content generated based on prompt."
+
+        return jsonify({
+            'success': True,
+            'video': generated_video,
+            'remaining_credits': new_credits
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+
+#----------Image Generation Endpoint ----------
+@app.route('/api/image/generate', methods=['POST'])
+def generate_video():
+    try:
+        auth_header = request.headers.get('Authorization', '')
+        if not auth_header.startswith('Bearer '):
+            return jsonify({'error': 'Authorization header missing or malformed'}), 401
+        token = auth_header.split(' ')[1]
+        uid = verify_token(token)
+        if not uid:
+            return jsonify({'error': 'Invalid token'}), 401
+
+        # Get user data and check credits
+        user_ref = db.reference(f'users/{uid}')
+        user_data = user_ref.get()
+        if not user_data or user_data.get('credits', 0) < 1:
+            return jsonify({'error': 'Insufficient credits'}), 400
+
+        # Deduct one credit
+        new_credits = user_data.get('credits', 0) - 1
+        user_ref.update({'credits': new_credits})
+
+        # Get prompt and generate video (simulation using Gemini)
+        req_data = request.get_json()
+        prompt = req_data.get('prompt', '')
+        if not prompt:
+            return jsonify({'error': 'Prompt is required'}), 400
+
+        model = configure_gemini()
+        # Call to Gemini to generate video content (this is a simulation)
+        response = model.generate(prompt=prompt)
+        # For demonstration, we assume the response contains a 'result' field
+        generated_video = response.result if hasattr(response, 'result') else "Video content generated based on prompt."
+
+        return jsonify({
+            'success': True,
+            'video': generated_video,
+            'remaining_credits': new_credits
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+# ---------- Preview Generation Endpoint ----------
+@app.route('/api/preview/generate', methods=['POST'])
 def generate_video():
     try:
         auth_header = request.headers.get('Authorization', '')