Update app.py

app.py

@@ -8,44 +8,40 @@ import cohere
 import requests
 import uuid
 from datetime import datetime
-import base64
 
-# Logging setup
+# Setup logging
 logging.basicConfig(level=logging.INFO)
 
-# SQLAlchemy base class
 class Base(DeclarativeBase):
     pass
 
-# Flask app
-app = Flask(__name__)
+# ✅ Use writable instance_path to avoid PermissionError
+app = Flask(__name__, instance_path=os.path.join(os.getcwd(), 'instance'))
 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:///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)
 
-# Login manager setup
+# Flask-Login 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 and clients
+# API keys
 API_KEY = os.environ.get("Api_key")
 HUGGINGFACE_API_KEY = os.environ.get("HUGGINGFACE_API_KEY")
-
-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
+cohere_client = cohere.Client(API_KEY) if API_KEY else None
+
 @login_manager.user_loader
 def load_user(user_id):
     from models import User
@@ -70,11 +66,13 @@ 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'])
@@ -85,10 +83,13 @@ 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,21 +102,19 @@ 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
 
     if not cohere_client:
-        return jsonify({'error': 'Cohere API key not configured'}), 500
+        return jsonify({'error': 'Cohere API not configured'}), 500
 
-    try:
-        story_prompt = f"""Transform this scene idea into a vivid, poetic paragraph with beautiful imagery:
+    story_prompt = f"""Transform this scene idea into a vivid paragraph:
 Scene idea: {scene_idea}"""
 
+    try:
         story_response = cohere_client.generate(
             model='command',
             prompt=story_prompt,
@@ -123,49 +122,40 @@ Scene idea: {scene_idea}"""
             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()
-
-        return jsonify({
-            'success': True,
-            'story': expanded_story,
-            'image_url': image_url,
-            'creation_id': creation.id
-        })
+    except Exception as e:
+        logging.error(f"Error generating story: {e}")
+        return jsonify({'error': 'Story generation failed'}), 500
 
+    image_url = None
+    try:
+        image_url = generate_image_hf(scene_idea, expanded_story)
     except Exception as e:
-        logging.error(f"❌ Error generating content: {e}")
-        return jsonify({'error': 'An error occurred during generation. Check logs.'}), 500
+        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})
 
 @app.route('/save_journal', methods=['POST'])
 @login_required
 def save_journal():
     from models import Creation
     data = request.get_json()
-    creation_id = data.get('creation_id')
-    journal_entry = data.get('journal_entry', '').strip()
-
+    creation_id, journal_entry = data.get('creation_id'), 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()
@@ -186,24 +176,30 @@ def get_creations():
     } for c in creations]})
 
 def generate_image_hf(scene_idea, story):
-    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
+    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
 from models import User, Creation
 with app.app_context():
     db.create_all()