Update app.py

app.py

@@ -1,14 +1,16 @@
-python
 import logging
 import os
 from logging.handlers import RotatingFileHandler
-from flask import Flask, render_template, request, redirect, url_for, flash, jsonify, Response
+
+from flask import (
+    Flask, render_template, request, redirect,
+    url_for, flash, jsonify, Response
+)
 from flask_sqlalchemy import SQLAlchemy
 from flask_migrate import Migrate
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy import or_
 import pandas as pd
-import openai, sys
 from dotenv import load_dotenv
 from openai import OpenAI
 from datetime import datetime, timedelta
@@ -18,30 +20,27 @@ load_dotenv()
 
 # ─── Logging configuration ────────────────────────────────────────────────────
 os.makedirs("/tmp/logs", exist_ok=True)
-
 file_handler = RotatingFileHandler(
     "/tmp/logs/app.log", maxBytes=10_000_000, backupCount=5
 )
 file_handler.setFormatter(logging.Formatter(
     "[%(asctime)s] %(levelname)s in %(module)s: %(message)s"
 ))
-
 console_handler = logging.StreamHandler()
-console_handler.setFormatter(logging.Formatter(
-    "[%(levelname)s] %(message)s"
-))
+console_handler.setFormatter(logging.Formatter("[%(levelname)s] %(message)s"))
 
 logging.basicConfig(level=logging.INFO, handlers=[console_handler, file_handler])
 
-# ─── Flask application setup ─────────────────────────────────────────────────
+# ─── Flask setup ───────────────────────────────────────────────────────────────
 app = Flask(__name__)
 app.logger.handlers = [console_handler, file_handler]
 app.logger.setLevel(logging.INFO)
 
-# ─── OpenAI client setup ─────────────────────────────────────────────────────
-openai.api_key = os.getenv("OPENAI_API_KEY")
-# or for the new v1 client:
-# client = OpenAI()
+# ─── OpenAI client setup ──────────────────────────────────────────────────────
+# client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))  # if you prefer v1 client
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OPENAI_API_KEY:
+    app.logger.warning("OPENAI_API_KEY is not set in the environment")
 
 # ─── Database configuration ───────────────────────────────────────────────────
 app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///jobs.db"
@@ -69,7 +68,6 @@ class Job(db.Model):
 
 # ─── Helpers ──────────────────────────────────────────────────────────────────
 def convert_excel_serial_dates():
-    """Convert Excel serial dates in `date_applied` to ISO strings."""
     updated = False
     for job in Job.query.all():
         da = job.date_applied
@@ -86,14 +84,13 @@ def convert_excel_serial_dates():
         app.logger.info("Converted Excel serial dates to ISO strings")
 
 def generate_interview_plan(job_description: str) -> str:
-    """Use OpenAI to generate an interview plan from the job description."""
     try:
-        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        client = OpenAI(api_key=OPENAI_API_KEY)
         resp = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content":
+                {"role": "user",   "content":
                     f"Create an interview plan for the following job description. "
                     f"Highlight key skills and requirements:\n{job_description}"
                 },
@@ -101,11 +98,11 @@ def generate_interview_plan(job_description: str) -> str:
             max_tokens=150
         )
         return resp.choices[0].message.content.strip()
-    except Exception as e:
-        app.logger.error("Error generating interview plan", exc_info=True)
+    except Exception:
+        app.logger.exception("Error generating interview plan")
         return "Error generating interview plan. Please try again later."
 
-# ─── App Initialization ──────────────────────────────────────────────────────
+# ─── Auto-initialize DB ────────────────────────────────────────────────────────
 @app.before_first_request
 def initialize_database():
     db.create_all()
@@ -135,9 +132,9 @@ def jobs():
             db.session.add(new_job)
             db.session.commit()
             return redirect(url_for("jobs"))
-        except SQLAlchemyError as e:
-            app.logger.error("Error adding job", exc_info=True)
-            flash(f"Error adding job: {e}")
+        except SQLAlchemyError:
+            app.logger.exception("Error adding job")
+            flash("Error adding job. Check logs for details.")
     # GET
     search = request.args.get("search", "").strip()
     status_filter = request.args.get("status_filter", "").strip()
@@ -161,8 +158,8 @@ def jobs():
 
     jobs_list = query.all()
     counts = {
-        "Applied": Job.query.filter_by(status="Applied").count(),
-        "Rejected": Job.query.filter_by(status="Rejected").count(),
+        "Applied":    Job.query.filter_by(status="Applied").count(),
+        "Rejected":   Job.query.filter_by(status="Rejected").count(),
         "Interviewing": Job.query.filter_by(status="Interviewing").count()
     }
     return render_template(
@@ -191,9 +188,9 @@ def edit_job(job_id):
             job.job_description = request.form.get("job_description")
             db.session.commit()
             return redirect(url_for("jobs"))
-        except SQLAlchemyError as e:
-            app.logger.error("Error updating job", exc_info=True)
-            flash(f"Error updating job: {e}")
+        except SQLAlchemyError:
+            app.logger.exception("Error updating job")
+            flash("Error updating job. Check logs for details.")
             db.session.rollback()
     return render_template("jobs.html", jobs=Job.query.all(), edit_job=job)
 
@@ -203,9 +200,9 @@ def delete_job(job_id):
     try:
         db.session.delete(job)
         db.session.commit()
-    except SQLAlchemyError as e:
-        app.logger.error("Error deleting job", exc_info=True)
-        flash(f"Error deleting job: {e}")
+    except SQLAlchemyError:
+        app.logger.exception("Error deleting job")
+        flash("Error deleting job. Check logs for details.")
         db.session.rollback()
     return redirect(url_for("jobs"))
 
@@ -216,7 +213,7 @@ def api_chat():
     if not msg:
         return jsonify(response="Please type something!")
     try:
-        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        client = OpenAI(api_key=OPENAI_API_KEY)
         resp = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
@@ -226,9 +223,9 @@ def api_chat():
             max_tokens=150
         )
         return jsonify(response=resp.choices[0].message.content.strip())
-    except Exception as e:
-        app.logger.error("Chat API error", exc_info=True)
-        return jsonify(response=f"Error: {e}")
+    except Exception:
+        app.logger.exception("Chat API error")
+        return jsonify(response="Error during chat. See logs.")
 
 @app.route("/prepme/<int:job_id>")
 def prepme(job_id):
@@ -239,27 +236,25 @@ def prepme(job_id):
         from docx import Document
         doc = Document(resume_path)
         resume_content = "\n".join(p.text for p in doc.paragraphs)
-    initial_context = f"Job Description:\n{job.job_description}\n\nResume:\n{resume_content}"
+    context = f"Job Description:\n{job.job_description}\n\nResume:\n{resume_content}"
     try:
-        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        client = OpenAI(api_key=OPENAI_API_KEY)
         resp = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": "You are a career coach."},
-                {"role": "user",   "content":
-                    f"Based on this context, help the user prepare:\n{initial_context}"
-                },
+                {"role": "user",   "content": context},
             ],
             max_tokens=150
         )
         initial_response = resp.choices[0].message.content.strip()
-    except Exception as e:
-        app.logger.error("PrepMe error", exc_info=True)
+    except Exception:
+        app.logger.exception("PrepMe error")
         initial_response = "Error generating prep response."
     return render_template(
         "prepme.html",
         job=job,
-        initial_context=initial_context,
+        initial_context=context,
         initial_response=initial_response
     )
 
@@ -302,4 +297,3 @@ def download_jobs():
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 5000))
     app.run(host="0.0.0.0", port=port, debug=False)
-