Update app.py

app.py

@@ -41,6 +41,7 @@ HUGGINGFACE_API_KEY = os.environ.get("HUGGINGFACE_API_KEY")
 HF_IMAGE_MODEL = "stabilityai/stable-diffusion-2-1"
 
 cohere_client = cohere.Client(API_KEY) if API_KEY else None
+logging.info(f"Received prompt: {scene_idea}")
 
 @login_manager.user_loader
 def load_user(user_id):
@@ -98,23 +99,33 @@ def logout():
     logout_user()
     return redirect(url_for('index'))
 
+
 @app.route('/generate', methods=['POST'])
 @login_required
 def generate_content():
     from models import Creation
+
+    # 1. ✅ Logging for debugging
+    logging.info("⚡ generate_content() called")
+
+    # 2. ✅ Extract prompt
     data = request.get_json()
     scene_idea = data.get('scene_idea', '').strip()
+    logging.info(f"📥 Received prompt: {scene_idea}")
 
     if not scene_idea:
+        logging.warning("❗ No scene idea provided")
         return jsonify({'error': 'Please provide a scene idea'}), 400
 
     if not cohere_client:
-        return jsonify({'error': 'Cohere API not configured'}), 500
-
-    story_prompt = f"""Transform this scene idea into a vivid paragraph:
-Scene idea: {scene_idea}"""
+        logging.error("❌ Cohere client is not configured. Check your API_KEY.")
+        return jsonify({'error': 'Text generation is unavailable'}), 500
 
+    # 3. ✅ Try generating story
     try:
+        story_prompt = f"""Transform this scene idea into a vivid paragraph:\nScene idea: {scene_idea}"""
+        logging.info("🧠 Sending prompt to Cohere")
+
         story_response = cohere_client.generate(
             model='command',
             prompt=story_prompt,
@@ -122,18 +133,25 @@ Scene idea: {scene_idea}"""
             temperature=0.7,
             k=0
         )
+
+        logging.info("✅ Cohere response received")
+
         expanded_story = story_response.generations[0].text.strip()
+        logging.info(f"📝 Generated story: {expanded_story[:60]}...")
+
     except Exception as e:
-        logging.error(f"Error generating story: {e}")
-        return jsonify({'error': 'Story generation failed'}), 500
+        logging.error(f"❌ Cohere generation error: {str(e)}")
+        return jsonify({'error': 'Failed to generate story'}), 500
 
+    # 4. ✅ Generate image (optional)
     image_url = None
     try:
         image_url = generate_image_hf(scene_idea, expanded_story)
+        logging.info(f"🖼️ Image generated: {image_url[:50]}...")
     except Exception as e:
-        image_url = None
-        logging.error(f"image generation completely failed")
+        logging.warning(f"⚠️ Image generation failed: {e}")
 
+    # 5. ✅ Save to DB
     creation_id = str(uuid.uuid4())
     creation = Creation(
         id=creation_id,
@@ -142,11 +160,16 @@ Scene idea: {scene_idea}"""
         story=expanded_story,
         image_url=image_url
     )
-
     db.session.add(creation)
     db.session.commit()
 
-    return jsonify({'success': True, 'story': expanded_story, 'image_url': image_url, 'creation_id': creation_id})
+    return jsonify({
+        'success': True,
+        'story': expanded_story,
+        'image_url': image_url,
+        'creation_id': creation_id
+    })
+
 
 @app.route('/save_journal', methods=['POST'])
 @login_required