Update app.py

app.py

@@ -6,9 +6,10 @@ import fitz  # PyMuPDF
 import io
 import requests
 import re
+import os
 
 # --- Config ---
-API_KEY = "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"  
+DEFAULT_API_KEY = "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"  
 API_URL = "https://openrouter.ai/api/v1/chat/completions"
 MODEL = "mistralai/mistral-7b-instruct"
 
@@ -66,6 +67,19 @@ st.markdown("""
         margin: 15px 0;
         border-radius: 0 8px 8px 0;
     }
+    .error-box {
+        background-color: #fdecea;
+        border-left: 5px solid #e74c3c;
+        padding: 15px;
+        margin: 15px 0;
+        border-radius: 0 8px 8px 0;
+    }
+    .api-container {
+        background-color: #eaf2f8;
+        border-radius: 10px;
+        padding: 15px;
+        margin: 15px 0;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -80,6 +94,15 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
+# API Key Setup
+st.markdown("---")
+st.markdown("### 🔑 API Configuration")
+st.info("You need an API key from [OpenRouter](https://openrouter.ai/) to use this tool. Get a free key and paste it below.")
+
+api_key = st.text_input("Enter your OpenRouter API Key:", type="password", 
+                       help="Get your API key from https://openrouter.ai/keys",
+                       value=os.getenv("OPENROUTER_API_KEY", DEFAULT_API_KEY))
+
 # Features in columns
 col1, col2, col3 = st.columns(3)
 with col1:
@@ -134,7 +157,7 @@ if uploaded_file:
     lab_text = st.text_area("", lab_text, height=300, label_visibility="collapsed")
 
 # --- AI Evaluation ---
-if lab_text.strip():
+if lab_text.strip() and api_key:
     # -- AI Evaluation Prompt --
     full_prompt = f"""You are a science teacher evaluating a student's lab report. Please provide a comprehensive analysis:
 
@@ -171,9 +194,9 @@ if lab_text.strip():
     Be concise but thorough in your analysis.
     """
 
-    def query_ai(prompt):
+    def query_ai(prompt, api_key):
         headers = {
-            "Authorization": f"Bearer {API_KEY}",
+            "Authorization": f"Bearer {api_key}",
             "Content-Type": "application/json"
         }
         payload = {
@@ -187,20 +210,23 @@ if lab_text.strip():
             response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
             response.raise_for_status()
             return response.json()['choices'][0]['message']['content']
+        except requests.exceptions.HTTPError as err:
+            st.error(f"API Error: {err.response.status_code} - {err.response.text}")
+            return None
         except Exception as e:
             st.error(f"Error connecting to AI service: {str(e)}")
             return None
 
     if st.button("🧪 Analyze Report", use_container_width=True):
         with st.spinner("🔍 Analyzing report with AI. This may take 20-30 seconds..."):
-            result = query_ai(full_prompt)
+            result = query_ai(full_prompt, api_key)
             
         if result:
             st.success("✅ Analysis Complete!")
             st.balloons()
             
             # Extract score using regex
-            score_match = re.search(r"Completeness Score:\s*(\d+)/10", result)
+            score_match = re.search(r"Completeness Score:\s*(\d+)/10", result, re.IGNORECASE)
             score = int(score_match.group(1)) if score_match else None
             
             # Display score in a card
@@ -305,11 +331,13 @@ if lab_text.strip():
             Answer the question based on the lab report. If the question can't be answered from the report, 
             suggest what information the student should add to answer it.
             """
-            followup_response = query_ai(followup_prompt)
+            followup_response = query_ai(followup_prompt, api_key)
         
         if followup_response:
             st.markdown("### 💬 AI Response")
             st.markdown(f'<div class="tip-box">{followup_response}</div>', unsafe_allow_html=True)
+elif lab_text.strip() and not api_key:
+    st.error("⚠️ Please enter a valid OpenRouter API key to analyze your report")
 else:
     # Show sample report if no file uploaded
     st.markdown("---")