Update app.py

app.py

@@ -6,13 +6,12 @@ import fitz  # PyMuPDF
 import io
 import requests
 import re
-import os
 from fpdf import FPDF
 from datetime import datetime
 
 # --- Config ---
-DEFAULT_API_KEY = "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"
 API_URL = "https://openrouter.ai/api/v1/chat/completions"
+API_KEY = "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"
 MODEL = "mistralai/mistral-7b-instruct"
 
 # Set page config
@@ -77,12 +76,6 @@ st.markdown("""
         margin: 15px 0;
         border-radius: 0 8px 8px 0;
     }
-    .api-container {
-        background-color: #eaf2f8;
-        border-radius: 10px;
-        padding: 15px;
-        margin: 15px 0;
-    }
     .experiment-card {
         background: linear-gradient(135deg, #f0f7ff, #e1effe);
         border-radius: 15px;
@@ -105,6 +98,12 @@ st.markdown("""
     [data-testid="stSidebar"] {
         background: linear-gradient(180deg, #e8f8f5, #d1f2eb) !important;
     }
+    .footer {
+        text-align: center;
+        padding: 20px;
+        color: #7f8c8d;
+        font-size: 14px;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -119,58 +118,26 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-# API Key Setup
-with st.sidebar:
-    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))
-    
-    st.markdown("---")
-    st.markdown("### 🧪 Experiment Templates")
-    st.caption("Quickly start with these pre-defined experiments:")
-    
-    # Experiment templates
-    experiments = {
-        "Vinegar + Baking Soda": {
-            "hypothesis": "Mixing vinegar and baking soda will produce bubbles due to a chemical reaction.",
-            "concept": "Acid-base reaction producing carbon dioxide."
-        },
-        "Floating Egg": {
-            "hypothesis": "An egg will float in salt water but sink in plain water.",
-            "concept": "Density difference between saltwater and freshwater."
-        },
-        "Lemon Battery": {
-            "hypothesis": "A lemon can produce electricity to power a small LED.",
-            "concept": "Chemical energy conversion to electrical energy."
-        }
+# Experiment templates
+experiments = {
+    "Vinegar + Baking Soda": {
+        "hypothesis": "Mixing vinegar and baking soda will produce bubbles due to a chemical reaction.",
+        "concept": "Acid-base reaction producing carbon dioxide."
+    },
+    "Floating Egg": {
+        "hypothesis": "An egg will float in salt water but sink in plain water.",
+        "concept": "Density difference between saltwater and freshwater."
+    },
+    "Lemon Battery": {
+        "hypothesis": "A lemon can produce electricity to power a small LED.",
+        "concept": "Chemical energy conversion to electrical energy."
     }
-    
-    selected_exp = st.selectbox("Choose an experiment template:", 
-                               list(experiments.keys()) + ["Custom Experiment"])
-    
-    st.markdown("---")
-    st.markdown("### 📘 Science Glossary Helper")
-    term = st.text_input("Enter a science term (e.g., osmosis, catalyst)")
-    if term:
-        with st.spinner("Looking up term..."):
-            ai_response = st.session_state.get('query_ai', lambda p: f"Explain '{term}'")(f"Explain the term '{term}' in simple words for a student.")
-        st.markdown(f"<div class='concept-box'>{ai_response}</div>", unsafe_allow_html=True)
-
-# Navigation
-app_mode = st.radio("Choose Mode:", ["🧪 Experiment Assistant", "📝 Lab Report Analyzer"], 
-                    horizontal=True, label_visibility="collapsed")
+}
 
 # AI Query Function
-def query_ai(prompt, api_key):
-    if not api_key:
-        st.error("⚠️ Please enter a valid API key")
-        return None
-        
+def query_ai(prompt):
     headers = {
-        "Authorization": f"Bearer {api_key}",
+        "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
     payload = {
@@ -192,9 +159,26 @@ def query_ai(prompt, api_key):
         st.error(f"Error connecting to AI service: {str(e)}")
         return None
 
-# Store query_ai in session state for glossary helper
-if 'query_ai' not in st.session_state:
-    st.session_state.query_ai = query_ai
+# Navigation
+app_mode = st.radio("Choose Mode:", ["🧪 Experiment Assistant", "📝 Lab Report Analyzer"], 
+                    horizontal=True, label_visibility="collapsed")
+
+# Sidebar
+with st.sidebar:
+    st.markdown("### 🧪 Experiment Templates")
+    st.caption("Quickly start with these pre-defined experiments:")
+    
+    selected_exp = st.selectbox("Choose an experiment template:", 
+                               list(experiments.keys()) + ["Custom Experiment"])
+    
+    st.markdown("---")
+    st.markdown("### 📘 Science Glossary Helper")
+    term = st.text_input("Enter a science term (e.g., osmosis, catalyst)")
+    if term:
+        with st.spinner("Looking up term..."):
+            ai_response = query_ai(f"Explain the term '{term}' in simple words for a student.")
+        if ai_response:
+            st.markdown(f"<div class='concept-box'>{ai_response}</div>", unsafe_allow_html=True)
 
 # --- Experiment Assistant Section ---
 if app_mode == "🧪 Experiment Assistant":
@@ -244,7 +228,7 @@ if app_mode == "🧪 Experiment Assistant":
             5. How to interpret the results
             """
             
-            explanation = query_ai(prompt, api_key)
+            explanation = query_ai(prompt)
         
         if explanation:
             st.success("✅ Experiment Guide Generated!")
@@ -377,7 +361,7 @@ else:
         lab_text = st.text_area("", lab_text, height=300, label_visibility="collapsed")
 
     # --- AI Evaluation ---
-    if lab_text.strip() and api_key:
+    if lab_text.strip():
         # -- AI Evaluation Prompt --
         full_prompt = f"""You are a science teacher evaluating a student's lab report. Please provide a comprehensive analysis:
 
@@ -416,7 +400,7 @@ else:
 
         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, api_key)
+                result = query_ai(full_prompt)
                 
             if result:
                 st.success("✅ Analysis Complete!")
@@ -528,13 +512,11 @@ else:
                 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, api_key)
+                followup_response = query_ai(followup_prompt)
             
             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("---")
@@ -554,10 +536,6 @@ else:
         
         st.info("👆 Upload your own lab report to get a personalized analysis!")
 
-# --- Feedback Section ---
+# Footer
 st.markdown("---")
-with st.expander("💬 Send Feedback"):
-    st.markdown("We'd love to hear your thoughts to improve this tool!")
-    feedback = st.text_area("What can we do better?")
-    if st.button("Submit Feedback", use_container_width=True):
-        st.success("Thank you for your feedback! We'll use it to improve the Science Lab Assistant.")
+st.markdown('<div class="footer">🔬 Science Lab Assistant | Made for Students & Educators</div>', unsafe_allow_html=True)