Update app.py

app.py

@@ -6,14 +6,24 @@ import fitz  # PyMuPDF
 import io
 import requests
 import re
-import numpy as np
+import os
 from fpdf import FPDF
 from datetime import datetime
-import os
+
+# --- Load Environment Variables ---
+from dotenv import load_dotenv
+load_dotenv()  # Load environment variables from .env file
 
 # --- Config ---
 API_URL = "https://openrouter.ai/api/v1/chat/completions"
-API_KEY = "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"
+
+# Retrieve API key from Streamlit secrets or environment variables
+try:
+    OPENROUTER_API_KEY = st.secrets["OPENROUTER_API_KEY"]
+except (KeyError, FileNotFoundError):
+    # Fallback for local development
+    OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
+
 MODEL = "mistralai/mistral-7b-instruct"
 
 # Set Tesseract path for different environments