Update app.py

app.py

@@ -8,39 +8,44 @@ import cohere
 import requests
 import uuid
 from datetime import datetime
+import base64
 
-# Setup logging
+# Logging setup
 logging.basicConfig(level=logging.INFO)
 
+# SQLAlchemy base class
 class Base(DeclarativeBase):
     pass
 
-app = Flask(__name__,instance_path="/tmp")
+# Flask app
+app = Flask(__name__)
 app.secret_key = os.environ.get("SESSION_SECRET", os.urandom(24).hex())
 
 # Database config
-app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("DATABASE_URL", "sqlite:////tmp/inkboard.db")
+app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("DATABASE_URL", "sqlite:///inkboard.db")
 app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
-app.config["SQLALCHEMY_ENGINE_OPTIONS"] = {
-    'pool_pre_ping': True,
-    "pool_recycle": 300,
-}
+app.config["SQLALCHEMY_ENGINE_OPTIONS"] = {'pool_pre_ping': True, "pool_recycle": 300}
 
 db = SQLAlchemy(app, model_class=Base)
 
-# Flask-Login setup
+# Login manager setup
 login_manager = LoginManager()
 login_manager.init_app(app)
 login_manager.login_view = 'login'
-login_manager.login_message = 'Please log in to access InkBoard'
 
-# API keys
-API_KEY = os.environ.get("API_KEY")
+# API keys and clients
+API_KEY = os.environ.get("Api_key")
 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
+if not API_KEY:
+    logging.warning("⚠️  Missing Cohere API key!")
+    cohere_client = None
+else:
+    cohere_client = cohere.Client(API_KEY)
+
+HF_IMAGE_MODEL = "stabilityai/stable-diffusion-2-1"
 
+# Load user from session
 @login_manager.user_loader
 def load_user(user_id):
     from models import User
@@ -65,13 +70,11 @@ def register():
 
         user = User(username=username, email=email)
         user.set_password(password)
-
         db.session.add(user)
         db.session.commit()
         login_user(user)
 
         return jsonify({'success': True, 'redirect': url_for('index')}) if request.is_json else redirect(url_for('index'))
-
     return render_template('register.html')
 
 @app.route('/login', methods=['GET', 'POST'])
@@ -82,13 +85,10 @@ def login():
         username, password = data.get('username', '').strip(), data.get('password', '').strip()
 
         user = User.query.filter((User.username == username) | (User.email == username)).first()
-
         if user and user.check_password(password):
             login_user(user)
             return jsonify({'success': True, 'redirect': url_for('index')}) if request.is_json else redirect(url_for('index'))
-
         return jsonify({'error': 'Invalid credentials'}), 401
-
     return render_template('login.html')
 
 @app.route('/logout')
@@ -101,54 +101,71 @@ def logout():
 @login_required
 def generate_content():
     from models import Creation
+
     data = request.get_json()
     scene_idea = data.get('scene_idea', '').strip()
+    logging.info(f"🎨 New prompt: {scene_idea}")
 
     if not scene_idea:
         return jsonify({'error': 'Please provide a scene idea'}), 400
 
-    story_prompt = f"""Transform this scene idea into a vivid paragraph:
+    if not cohere_client:
+        return jsonify({'error': 'Cohere API key not configured'}), 500
+
+    try:
+        story_prompt = f"""Transform this scene idea into a vivid, poetic paragraph with beautiful imagery:
 Scene idea: {scene_idea}"""
 
-    story_response = cohere_client.generate(
-        model='command',
-        prompt=story_prompt,
-        max_tokens=200,
-        temperature=0.7,
-        k=0
-    )
+        story_response = cohere_client.generate(
+            model='command',
+            prompt=story_prompt,
+            max_tokens=200,
+            temperature=0.7,
+            k=0
+        )
+
+        expanded_story = story_response.generations[0].text.strip()
+        logging.info(f"📝 Story generated: {expanded_story[:60]}...")
+
+        image_url = None
+        if HUGGINGFACE_API_KEY:
+            image_url = generate_image_hf(scene_idea, expanded_story)
+        else:
+            logging.warning("⚠️  Missing HuggingFace API key. No image will be generated.")
+
+        creation = Creation(
+            id=str(uuid.uuid4()),
+            user_id=current_user.id,
+            scene_idea=scene_idea,
+            story=expanded_story,
+            image_url=image_url
+        )
+        db.session.add(creation)
+        db.session.commit()
 
-    expanded_story = story_response.generations[0].text.strip()
+        return jsonify({
+            'success': True,
+            'story': expanded_story,
+            'image_url': image_url,
+            'creation_id': creation.id
+        })
 
-    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()
-
-    return jsonify({'success': True, 'story': expanded_story, 'image_url': image_url, 'creation_id': creation_id})
+        logging.error(f"❌ Error generating content: {e}")
+        return jsonify({'error': 'An error occurred during generation. Check logs.'}), 500
 
 @app.route('/save_journal', methods=['POST'])
 @login_required
 def save_journal():
     from models import Creation
     data = request.get_json()
-    creation_id, journal_entry = data.get('creation_id'), data.get('journal_entry', '').strip()
+    creation_id = data.get('creation_id')
+    journal_entry = data.get('journal_entry', '').strip()
+
     creation = Creation.query.filter_by(id=creation_id, user_id=current_user.id).first()
     if not creation:
         return jsonify({'error': 'Not found'}), 404
+
     creation.journal_entry = journal_entry
     creation.updated_at = datetime.utcnow()
     db.session.commit()
@@ -169,30 +186,24 @@ def get_creations():
     } for c in creations]})
 
 def generate_image_hf(scene_idea, story):
-    prompt = f"An artistic illustration of: {scene_idea}, dreamy and vivid"
-    for model in [
-        "runwayml/stable-diffusion-v1-5",
-        "stabilityai/stable-diffusion-2-1",
-        "CompVis/stable-diffusion-v1-4"
-    ]:
-        try:
-            res = requests.post(
-                f"https://api-inference.huggingface.co/models/{model}",
-                headers={"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"},
-                json={"inputs": prompt},
-                timeout=60
-            )
-            if res.status_code == 200:
-                import base64
-                return f"data:image/png;base64,{base64.b64encode(res.content).decode()}"
-        except Exception as e:
-            continue
-    return generate_svg_placeholder(scene_idea, story)
-
-def generate_svg_placeholder(scene_idea, story):
-    return "<svg>Your beautiful fallback SVG here</svg>"
-
-# DB setup
+    prompt = f"An artistic illustration of: {scene_idea}, dreamy, vivid, high detail"
+    model = HF_IMAGE_MODEL
+    try:
+        response = requests.post(
+            f"https://api-inference.huggingface.co/models/{model}",
+            headers={"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"},
+            json={"inputs": prompt},
+            timeout=60
+        )
+        if response.status_code == 200:
+            return f"data:image/png;base64,{base64.b64encode(response.content).decode()}"
+        else:
+            logging.warning(f"Model failed: {response.status_code} - {response.text}")
+    except Exception as e:
+        logging.error(f"Exception generating image: {e}")
+    return None  # fallback
+
+# Import models and initialize DB
 from models import User, Creation
 with app.app_context():
     db.create_all()