Update app.py

app.py

@@ -2,19 +2,25 @@
 import streamlit as st
 import pytesseract
 from PIL import Image
-import fitz  # PyMuPDF
 import io
 import requests
 import re
-import numpy as np
 from fpdf import FPDF
 from datetime import datetime
 import os
+from dotenv import load_dotenv
+import pdfplumber  # Alternative PDF library
 
-# --- Config ---
-API_URL = "https://api.groq.com/openai/v1/chat/completions"
-# Retrieve API key from environment variable
+# Load environment variables
+load_dotenv()
+
+# --- Retrieve API Key ---
 API_KEY = os.getenv("OPENROUTER_API_KEY")
+if not API_KEY:
+    st.error("OPENROUTER_API_KEY not found in environment variables! Please add it to your .env file.")
+    st.stop()
+
+API_URL = "https://api.groq.com/openai/v1/chat/completions"
 MODEL = "llama3-70b-8192"  # Or "mixtral-8x7b-32768" or "gemma-7b-it"
 
 # Set Tesseract path for different environments
@@ -22,7 +28,7 @@ try:
     # For Windows
     if os.name == 'nt':
         pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
-    # For Linux (Streamlit Sharing)
+    # For Linux
     elif 'tesseract' not in os.environ.get('PATH', ''):
         pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'
 except Exception as e:
@@ -156,6 +162,7 @@ experiments = {
 }
 
 # AI Query Function
+# Replace the query_ai function with this version
 def query_ai(prompt):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -171,6 +178,21 @@ def query_ai(prompt):
     }
     try:
         response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
+        
+        # Check for 401 Unauthorized specifically
+        if response.status_code == 401:
+            st.error("""
+            **Invalid API Key!**
+            
+            Your Groq API key is either:
+            - Missing
+            - Incorrect
+            - Expired
+            
+            Please check your .env file and ensure you have a valid key.
+            """)
+            return None
+            
         response.raise_for_status()
         return response.json()['choices'][0]['message']['content']
     except requests.exceptions.HTTPError as err:
@@ -179,7 +201,6 @@ def query_ai(prompt):
     except Exception as e:
         st.error(f"Error connecting to AI service: {str(e)}")
         return None
-
 # Navigation
 app_mode = st.radio("Choose Mode:", ["🧪 Experiment Assistant", "📝 Lab Report Analyzer"], 
                     horizontal=True, label_visibility="collapsed")
@@ -372,9 +393,9 @@ else:
 
         if file_ext == "pdf":
             try:
-                doc = fitz.open(stream=file_bytes, filetype="pdf")
-                for page in doc:
-                    lab_text += page.get_text()
+                with pdfplumber.open(io.BytesIO(file_bytes)) as pdf:
+                    for page in pdf.pages:
+                        lab_text += page.extract_text() + "\n"
                 st.success("✅ PDF text extracted successfully!")
             except Exception as e:
                 st.error(f"Error reading PDF: {str(e)}")