Update src/pages/categorized/page6.py

src/pages/categorized/page6.py

@@ -4,7 +4,7 @@ import json
 import tempfile
 import zipfile
 from io import BytesIO
-import fitz  # PyMuPDF
+import fitz  
 import cv2
 import numpy as np
 
@@ -15,7 +15,11 @@ import base64
 from typing import Dict, Any, Optional
 from collections import defaultdict
 
-API_KEY = "AIzaSyB5RZzjRHaFeGYyMf0WuGJ65RHfhrJ2vtw"
+API_KEY = os.environ.get("GEMINI_API_KEY")
+if not API_KEY:
+    st.error("Gemini API key not found. Set GEMINI_API_KEY in Hugging Face Secrets.")
+    st.stop()
+
 API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-preview-09-2025:generateContent?key={API_KEY}"
 
 SCHEMA = {
@@ -88,7 +92,11 @@ def call_gemini_from_bytes(pdf_bytes: bytes, filename: str) -> Optional[Dict[str
     }
 
     try:
-        r = requests.post(API_URL, json=payload, timeout=300)
+        headers = {
+            "Content-Type": "application/json"
+        }
+        
+        r = requests.post(API_URL, json=payload, headers=headers, timeout=300)
         r.raise_for_status()
         data = r.json()