Update app.py

app.py

@@ -108,24 +108,23 @@ def logout():
     logout_user()
     return redirect(url_for('index'))
 
-
 @app.route('/generate', methods=['POST'])
 @login_required
 def generate_content():
-    logging.info("⚡ generate_content() called")
-    data = request.get_json()
-    scene_idea = data.get('scene_idea', '').strip()
-    logging.info(f"📥 Received prompt: {scene_idea}")
-
-    if not scene_idea:
-        return jsonify({'error': 'Please provide a scene idea'}), 400
+    from models import Creation
 
     if not cohere_client:
-        logging.error("❌ Cohere client not configured")
-        return jsonify({'error': 'Text generation unavailable'}), 500
+        return jsonify({'error': 'Cohere API key not configured'}), 500
 
     try:
-        story_prompt = f"""Transform this scene idea into a vivid paragraph:\nScene idea: {scene_idea}"""
+        data = request.get_json()
+        scene_idea = data.get('scene_idea', '').strip()
+
+        if not scene_idea:
+            return jsonify({'error': 'Please provide a scene idea'}), 400
+
+        story_prompt = f"Transform this scene idea into a vivid paragraph:\nScene idea: {scene_idea}"
+
         story_response = cohere_client.generate(
             model='command',
             prompt=story_prompt,
@@ -133,35 +132,37 @@ def generate_content():
             temperature=0.7,
             k=0
         )
+
         expanded_story = story_response.generations[0].text.strip()
-        logging.info("✅ Story generated")
-    except Exception as e:
-        logging.error(f"❌ Cohere generation error: {str(e)}")
-        return jsonify({'error': 'Failed to generate story'}), 500
 
-    image_url = None
-    try:
-        image_url = generate_image_hf(scene_idea, expanded_story)
-    except Exception as e:
-        logging.warning(f"⚠️ Image generation failed: {e}")
-
-    creation_id = str(uuid.uuid4())
-    creation = Creation(
-        id=creation_id,
-        user_id=current_user.id,
-        scene_idea=scene_idea,
-        story=expanded_story,
-        image_url=image_url
-    )
-    db.session.add(creation)
-    db.session.commit()
+        try:
+            image_url = generate_image_hf(scene_idea, expanded_story)
+        except Exception as e:
+            logging.warning(f"Image generation failed: {e}")
+            image_url = generate_svg_placeholder(scene_idea, expanded_story)
+
+        creation_id = str(uuid.uuid4())
+        creation = Creation(
+            id=creation_id,
+            user_id=current_user.id,
+            scene_idea=scene_idea,
+            story=expanded_story,
+            image_url=image_url
+        )
 
-    return jsonify({
-        'success': True,
-        'story': expanded_story,
-        'image_url': image_url,
-        'creation_id': creation_id
-    })
+        db.session.add(creation)
+        db.session.commit()
+
+        return jsonify({
+            'success': True,
+            'story': expanded_story,
+            'image_url': image_url,
+            'creation_id': creation_id
+        })
+
+    except Exception as e:
+        logging.error(f"Error in /generate route: {e}")
+        return jsonify({'error': 'Server failed to generate content'}), 500
 
 
 @app.route('/save_journal', methods=['POST'])