Update app.py

app.py

@@ -7,17 +7,20 @@ import io
 import requests
 import re
 import os
+from fpdf import FPDF
+from datetime import datetime
 
 # --- Config ---
-DEFAULT_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"
 
 # Set page config
 st.set_page_config(
-    page_title="🔬 AI Science Lab Assistant", 
+    page_title="🔬 Science Lab Assistant", 
     layout="centered",
-    page_icon="🔬"
+    page_icon="🔬",
+    initial_sidebar_state="expanded"
 )
 
 # Custom CSS for styling
@@ -80,279 +83,481 @@ st.markdown("""
         padding: 15px;
         margin: 15px 0;
     }
+    .experiment-card {
+        background: linear-gradient(135deg, #f0f7ff, #e1effe);
+        border-radius: 15px;
+        padding: 20px;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+        margin-bottom: 20px;
+        transition: all 0.3s;
+    }
+    .experiment-card:hover {
+        transform: translateY(-5px);
+        box-shadow: 0 6px 12px rgba(0,0,0,0.15);
+    }
+    .concept-box {
+        background-color: #ebf5fb;
+        border-left: 5px solid #3498db;
+        padding: 15px;
+        margin: 15px 0;
+        border-radius: 0 8px 8px 0;
+    }
+    [data-testid="stSidebar"] {
+        background: linear-gradient(180deg, #e8f8f5, #d1f2eb) !important;
+    }
     </style>
     """, unsafe_allow_html=True)
 
 # Header
-st.markdown('<p class="header">🔬 AI Science Lab Assistant</p>', unsafe_allow_html=True)
+st.markdown('<p class="header">🔬 Science Lab Assistant</p>', unsafe_allow_html=True)
 
 # Introduction
 st.markdown("""
 <div style="text-align: center; margin-bottom: 30px;">
-    <p style="font-size: 18px;">Transform your lab reports with AI-powered analysis! Get instant feedback on completeness, 
-    receive improvement suggestions, and ask questions about your scientific work.</p>
+    <p style="font-size: 18px;">Your all-in-one science companion! Design experiments, generate reports, 
+    and get AI-powered feedback on your lab work.</p>
 </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:
-    st.markdown("""
-        <div style="text-align: center;">
-            <h4>🔍 Comprehensive Analysis</h4>
-            <p>Checks for all essential lab report sections</p>
-        </div>
-    """, unsafe_allow_html=True)
+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.")
     
-with col2:
-    st.markdown("""
-        <div style="text-align: center;">
-            <h4>💯 Smart Scoring</h4>
-            <p>Grades your report on completeness and structure</p>
-        </div>
-    """, unsafe_allow_html=True)
+    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))
     
-with col3:
-    st.markdown("""
-        <div style="text-align: center;">
-            <h4>📈 Improvement Tips</h4>
-            <p>Personalized suggestions to enhance your report</p>
-        </div>
-    """, unsafe_allow_html=True)
-
-# Divider
-st.markdown("---")
-
-# --- File Upload ---
-st.markdown('<p class="subheader">📤 Upload Your Lab Report</p>', unsafe_allow_html=True)
-uploaded_file = st.file_uploader("Upload image (JPG, PNG) or PDF", 
-                                 type=["jpg", "jpeg", "png", "pdf"],
-                                 label_visibility="collapsed")
-
-lab_text = ""
-if uploaded_file:
-    file_bytes = uploaded_file.read()
-    file_ext = uploaded_file.name.split(".")[-1].lower()
-
-    if file_ext == "pdf":
-        doc = fitz.open(stream=file_bytes, filetype="pdf")
-        for page in doc:
-            lab_text += page.get_text()
-    else:
-        image = Image.open(io.BytesIO(file_bytes))
-        lab_text = pytesseract.image_to_string(image)
+    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."
+        }
+    }
+    
+    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)
 
-    # Allow text editing
-    st.markdown('<p class="subheader">✍️ Extracted Text</p>', unsafe_allow_html=True)
-    st.caption("Review and edit the extracted text if needed before analysis")
-    lab_text = st.text_area("", lab_text, height=300, label_visibility="collapsed")
+# Navigation
+app_mode = st.radio("Choose Mode:", ["🧪 Experiment Assistant", "📝 Lab Report Analyzer"], 
+                    horizontal=True, label_visibility="collapsed")
 
-# --- AI Evaluation ---
-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:
+# AI Query Function
+def query_ai(prompt, api_key):
+    if not api_key:
+        st.error("⚠️ Please enter a valid API key")
+        return None
+        
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "model": MODEL,
+        "messages": [
+            {"role": "system", "content": "You are a helpful science teacher providing detailed explanations."},
+            {"role": "user", "content": prompt}
+        ],
+        "temperature": 0.7
+    }
+    try:
+        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
 
-    Lab Report:
-    {lab_text}
+# Store query_ai in session state for glossary helper
+if 'query_ai' not in st.session_state:
+    st.session_state.query_ai = query_ai
 
-    Evaluation Guidelines:
-    1. **Section Check**: Identify which of these sections are present and which are missing:
-        - Title
-        - Objective
-        - Hypothesis
-        - Materials
-        - Procedure
-        - Observations
-        - Results
-        - Conclusion
-        - References
-    
-    2. **Completeness Score**: 
-        - Assign a numerical score from 1-10 based on completeness
-        - Justify the score based on missing sections and content quality
+# --- Experiment Assistant Section ---
+if app_mode == "🧪 Experiment Assistant":
+    st.markdown('<p class="subheader">🔍 Design Your Experiment</p>', unsafe_allow_html=True)
     
-    3. **Improvement Tips**:
-        - For each missing section, explain why it's important
-        - Provide specific suggestions for improvement (e.g., "Try writing a more detailed observation section by including quantitative data")
-        - Highlight any sections that need more detail or clarity
-    
-    4. **Structure Response**:
-        - Start with: "### Missing Sections:"
-        - Then: "### Completeness Score: X/10"
-        - Then: "### Improvement Tips:"
-        - Finally: "### Detailed Feedback:"
+    with st.form("experiment_form"):
+        # Pre-fill if template selected
+        if selected_exp != "Custom Experiment" and selected_exp in experiments:
+            default_hypo = experiments[selected_exp]["hypothesis"]
+            concept = experiments[selected_exp]["concept"]
+            exp_name = selected_exp
+        else:
+            default_hypo = ""
+            concept = ""
+            exp_name = st.text_input("Experiment Name", placeholder="e.g., Effect of Temperature on Enzyme Activity")
+        
+        hypo = st.text_area("Your Hypothesis", value=default_hypo, 
+                           placeholder="What do you predict will happen?")
+        
+        materials = st.text_area("Materials Needed", 
+                               placeholder="List all materials needed for this experiment")
+        
+        procedure = st.text_area("Procedure Steps", 
+                               placeholder="Step-by-step instructions for conducting the experiment")
+        
+        submit = st.form_submit_button("🔍 Generate Experiment Guide", use_container_width=True)
     
-    Be concise but thorough in your analysis.
-    """
-
-    def query_ai(prompt, api_key):
-        headers = {
-            "Authorization": f"Bearer {api_key}",
-            "Content-Type": "application/json"
-        }
-        payload = {
-            "model": MODEL,
-            "messages": [
-                {"role": "system", "content": "You are a helpful science teacher providing detailed lab report feedback."},
-                {"role": "user", "content": prompt}
-            ]
-        }
-        try:
-            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, 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, re.IGNORECASE)
-            score = int(score_match.group(1)) if score_match else None
-            
-            # Display score in a card
-            if score is not None:
-                with st.container():
-                    st.markdown('<div class="score-card">', unsafe_allow_html=True)
-                    
-                    # Create columns for score visualization
-                    col1, col2 = st.columns([1, 3])
-                    
-                    with col1:
-                        st.markdown(f"<h2 style='text-align: center; color: #28b463;'>{score}/10</h2>", 
-                                   unsafe_allow_html=True)
-                        st.markdown("<h4 style='text-align: center;'>Completeness Score</h4>", 
-                                  unsafe_allow_html=True)
-                    
-                    with col2:
-                        # Create a color gradient based on score
-                        if score >= 8:
-                            color = "#28b463"  # Green
-                        elif score >= 5:
-                            color = "#f39c12"  # Orange
-                        else:
-                            color = "#e74c3c"  # Red
-                            
-                        # Display progress bar with styling
-                        st.progress(score/10, text=f"{score*10}% complete")
-                        st.markdown(
-                            f"<style>"
-                            f".stProgress > div > div > div {{"
-                            f"    background-color: {color} !important;"
-                            f"    border-radius: 10px;"
-                            f"}}"
-                            f"</style>",
-                            unsafe_allow_html=True
-                        )
-                    
-                    st.markdown('</div>', unsafe_allow_html=True)
-            
-            # Display AI analysis with formatting
-            st.markdown("## 📝 Analysis Results")
+    if submit:
+        if not exp_name or not hypo:
+            st.warning("Please provide at least an experiment name and hypothesis")
+            st.stop()
             
-            # Split sections for better display
-            sections = {
-                "Missing Sections": None,
-                "Improvement Tips": None,
-                "Detailed Feedback": None
-            }
+        with st.spinner("Designing your experiment guide..."):
+            prompt = f"""
+            Create a comprehensive guide for a science experiment with the following details:
             
-            current_section = None
-            for line in result.split('\n'):
-                if "### Missing Sections:" in line:
-                    current_section = "Missing Sections"
-                    sections[current_section] = []
-                elif "### Improvement Tips:" in line:
-                    current_section = "Improvement Tips"
-                    sections[current_section] = []
-                elif "### Detailed Feedback:" in line:
-                    current_section = "Detailed Feedback"
-                    sections[current_section] = []
-                elif current_section and line.strip():
-                    sections[current_section].append(line)
+            Experiment Name: {exp_name}
+            Hypothesis: {hypo}
+            Materials: {materials if materials else 'Not specified'}
+            Procedure: {procedure if procedure else 'Not specified'}
             
-            # Display each section
-            if sections["Missing Sections"]:
-                st.markdown("### 🔍 Missing Sections")
-                missing_text = '\n'.join(sections["Missing Sections"])
-                st.markdown(f'<div class="highlight">{missing_text}</div>', unsafe_allow_html=True)
+            Please provide:
+            1. A clear explanation of the scientific concept behind the experiment
+            2. Step-by-step instructions for conducting the experiment
+            3. Safety precautions
+            4. Expected results and why they're expected
+            5. How to interpret the results
+            """
             
-            if sections["Improvement Tips"]:
-                st.markdown("### 💡 Improvement Tips")
-                tips_text = '\n'.join(sections["Improvement Tips"])
-                st.markdown(f'<div class="tip-box">{tips_text}</div>', unsafe_allow_html=True)
+            explanation = query_ai(prompt, api_key)
+        
+        if explanation:
+            st.success("✅ Experiment Guide Generated!")
+            st.balloons()
             
-            if sections["Detailed Feedback"]:
-                st.markdown("### 📋 Detailed Feedback")
-                st.write('\n'.join(sections["Detailed Feedback"]))
+            # Display explanation
+            st.markdown("### 🧪 Experiment Guide")
+            st.markdown(f"<div class='tip-box'>{explanation}</div>", unsafe_allow_html=True)
             
-            # Show full AI response in expander
-            with st.expander("View Full AI Analysis"):
-                st.markdown(result)
+            # Generate PDF report
+            def generate_pdf_report(exp_name, hypo, explanation, materials, procedure):
+                pdf = FPDF()
+                pdf.add_page()
+                pdf.set_font("Arial", size=12)
+                
+                # Title
+                pdf.set_font("Arial", 'B', 16)
+                pdf.cell(200, 10, txt="Science Experiment Guide", ln=True, align='C')
+                pdf.ln(15)
+                
+                # Experiment details
+                pdf.set_font("Arial", 'B', 14)
+                pdf.cell(0, 10, txt=f"Experiment: {exp_name}", ln=True)
+                pdf.ln(5)
+                
+                pdf.set_font("Arial", 'B', 12)
+                pdf.cell(0, 10, txt="Hypothesis:", ln=True)
+                pdf.set_font("Arial", '', 12)
+                pdf.multi_cell(0, 8, txt=hypo)
+                pdf.ln(5)
+                
+                if materials:
+                    pdf.set_font("Arial", 'B', 12)
+                    pdf.cell(0, 10, txt="Materials:", ln=True)
+                    pdf.set_font("Arial", '', 12)
+                    pdf.multi_cell(0, 8, txt=materials)
+                    pdf.ln(5)
+                
+                if procedure:
+                    pdf.set_font("Arial", 'B', 12)
+                    pdf.cell(0, 10, txt="Procedure:", ln=True)
+                    pdf.set_font("Arial", '', 12)
+                    pdf.multi_cell(0, 8, txt=procedure)
+                    pdf.ln(10)
+                
+                pdf.set_font("Arial", 'B', 12)
+                pdf.cell(0, 10, txt="Experiment Guide:", ln=True)
+                pdf.set_font("Arial", '', 12)
+                pdf.multi_cell(0, 8, txt=explanation)
+                
+                filename = f"experiment_guide_{datetime.now().strftime('%Y%m%d%H%M%S')}.pdf"
+                pdf.output(filename)
+                return filename
             
-    # --- Question Answering Section ---
+            pdf_file = generate_pdf_report(exp_name, hypo, explanation, materials, procedure)
+            with open(pdf_file, "rb") as file:
+                st.download_button("📄 Download Experiment Guide (PDF)", file, 
+                                  file_name=f"{exp_name}_guide.pdf", 
+                                  use_container_width=True)
+    
+    # Experiment examples
     st.markdown("---")
-    st.markdown('<p class="subheader">❓ Ask About Your Report</p>', unsafe_allow_html=True)
+    st.markdown('<p class="subheader">🔬 Popular Science Experiments</p>', unsafe_allow_html=True)
     
-    col1, col2 = st.columns([3, 1])
+    col1, col2 = st.columns(2)
     with col1:
-        user_question = st.text_input("Ask a question about your lab report", 
-                                     placeholder="e.g., How can I improve my hypothesis?")
-    with col2:
-        st.markdown("<div style='height: 28px;'></div>", unsafe_allow_html=True)
-        ask_button = st.button("🔍 Ask Question", use_container_width=True)
+        with st.container():
+            st.markdown('<div class="experiment-card">', unsafe_allow_html=True)
+            st.markdown("#### 🧫 Vinegar + Baking Soda")
+            st.markdown("**Hypothesis:** Mixing vinegar and baking soda will produce bubbles due to a chemical reaction.")
+            st.markdown("**Concept:** Acid-base reaction producing carbon dioxide.")
+            if st.button("Try This Experiment", key="vinegar", use_container_width=True):
+                st.session_state.selected_exp = "Vinegar + Baking Soda"
+            st.markdown('</div>', unsafe_allow_html=True)
+            
+        with st.container():
+            st.markdown('<div class="experiment-card">', unsafe_allow_html=True)
+            st.markdown("#### 🥚 Floating Egg")
+            st.markdown("**Hypothesis:** An egg will float in salt water but sink in plain water.")
+            st.markdown("**Concept:** Density difference between saltwater and freshwater.")
+            if st.button("Try This Experiment", key="egg", use_container_width=True):
+                st.session_state.selected_exp = "Floating Egg"
+            st.markdown('</div>', unsafe_allow_html=True)
     
-    if (ask_button or user_question) and user_question.strip():
-        with st.spinner("Thinking..."):
-            followup_prompt = f"""Lab Report:
-            {lab_text}
+    with col2:
+        with st.container():
+            st.markdown('<div class="experiment-card">', unsafe_allow_html=True)
+            st.markdown("#### 🍋 Lemon Battery")
+            st.markdown("**Hypothesis:** A lemon can produce electricity to power a small LED.")
+            st.markdown("**Concept:** Chemical energy conversion to electrical energy.")
+            if st.button("Try This Experiment", key="lemon", use_container_width=True):
+                st.session_state.selected_exp = "Lemon Battery"
+            st.markdown('</div>', unsafe_allow_html=True)
             
-            Question: {user_question}
+        with st.container():
+            st.markdown('<div class="experiment-card">', unsafe_allow_html=True)
+            st.markdown("#### 🌈 Rainbow in a Glass")
+            st.markdown("**Hypothesis:** Different sugar solutions can form colorful layers in a glass.")
+            st.markdown("**Concept:** Density gradient formation.")
+            if st.button("Try This Experiment", key="rainbow", use_container_width=True):
+                st.session_state.selected_exp = "Custom Experiment"
+                st.session_state.custom_exp = "Rainbow in a Glass"
+                st.session_state.custom_hypo = "Different sugar solutions will form distinct layers based on their density."
+            st.markdown('</div>', unsafe_allow_html=True)
+
+# --- Lab Report Analyzer Section ---
+else:
+    # --- File Upload ---
+    st.markdown('<p class="subheader">📤 Upload Your Lab Report</p>', unsafe_allow_html=True)
+    uploaded_file = st.file_uploader("Upload image (JPG, PNG) or PDF", 
+                                     type=["jpg", "jpeg", "png", "pdf"],
+                                     label_visibility="collapsed")
+
+    lab_text = ""
+    if uploaded_file:
+        file_bytes = uploaded_file.read()
+        file_ext = uploaded_file.name.split(".")[-1].lower()
+
+        if file_ext == "pdf":
+            doc = fitz.open(stream=file_bytes, filetype="pdf")
+            for page in doc:
+                lab_text += page.get_text()
+        else:
+            image = Image.open(io.BytesIO(file_bytes))
+            lab_text = pytesseract.image_to_string(image)
+
+        # Allow text editing
+        st.markdown('<p class="subheader">✍️ Extracted Text</p>', unsafe_allow_html=True)
+        st.caption("Review and edit the extracted text if needed before analysis")
+        lab_text = st.text_area("", lab_text, height=300, label_visibility="collapsed")
+
+    # --- AI Evaluation ---
+    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:
+
+        Lab Report:
+        {lab_text}
+
+        Evaluation Guidelines:
+        1. **Section Check**: Identify which of these sections are present and which are missing:
+            - Title
+            - Objective
+            - Hypothesis
+            - Materials
+            - Procedure
+            - Observations
+            - Results
+            - Conclusion
+            - References
+        
+        2. **Completeness Score**: 
+            - Assign a numerical score from 1-10 based on completeness
+            - Justify the score based on missing sections and content quality
+        
+        3. **Improvement Tips**:
+            - For each missing section, explain why it's important
+            - Provide specific suggestions for improvement (e.g., "Try writing a more detailed observation section by including quantitative data")
+            - Highlight any sections that need more detail or clarity
+        
+        4. **Structure Response**:
+            - Start with: "### Missing Sections:"
+            - Then: "### Completeness Score: X/10"
+            - Then: "### Improvement Tips:"
+            - Finally: "### Detailed Feedback:"
+        
+        Be concise but thorough in your analysis.
+        """
+
+        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)
+                
+            if result:
+                st.success("✅ Analysis Complete!")
+                st.balloons()
+                
+                # Extract score using regex
+                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
+                if score is not None:
+                    with st.container():
+                        st.markdown('<div class="score-card">', unsafe_allow_html=True)
+                        
+                        # Create columns for score visualization
+                        col1, col2 = st.columns([1, 3])
+                        
+                        with col1:
+                            st.markdown(f"<h2 style='text-align: center; color: #28b463;'>{score}/10</h2>", 
+                                       unsafe_allow_html=True)
+                            st.markdown("<h4 style='text-align: center;'>Completeness Score</h4>", 
+                                      unsafe_allow_html=True)
+                        
+                        with col2:
+                            # Create a color gradient based on score
+                            if score >= 8:
+                                color = "#28b463"  # Green
+                            elif score >= 5:
+                                color = "#f39c12"  # Orange
+                            else:
+                                color = "#e74c3c"  # Red
+                                
+                            # Display progress bar with styling
+                            st.progress(score/10, text=f"{score*10}% complete")
+                            st.markdown(
+                                f"<style>"
+                                f".stProgress > div > div > div {{"
+                                f"    background-color: {color} !important;"
+                                f"    border-radius: 10px;"
+                                f"}}"
+                                f"</style>",
+                                unsafe_allow_html=True
+                            )
+                        
+                        st.markdown('</div>', unsafe_allow_html=True)
+                
+                # Display AI analysis with formatting
+                st.markdown("## 📝 Analysis Results")
+                
+                # Split sections for better display
+                sections = {
+                    "Missing Sections": None,
+                    "Improvement Tips": None,
+                    "Detailed Feedback": None
+                }
+                
+                current_section = None
+                for line in result.split('\n'):
+                    if "### Missing Sections:" in line:
+                        current_section = "Missing Sections"
+                        sections[current_section] = []
+                    elif "### Improvement Tips:" in line:
+                        current_section = "Improvement Tips"
+                        sections[current_section] = []
+                    elif "### Detailed Feedback:" in line:
+                        current_section = "Detailed Feedback"
+                        sections[current_section] = []
+                    elif current_section and line.strip():
+                        sections[current_section].append(line)
+                
+                # Display each section
+                if sections["Missing Sections"]:
+                    st.markdown("### 🔍 Missing Sections")
+                    missing_text = '\n'.join(sections["Missing Sections"])
+                    st.markdown(f'<div class="highlight">{missing_text}</div>', unsafe_allow_html=True)
+                
+                if sections["Improvement Tips"]:
+                    st.markdown("### 💡 Improvement Tips")
+                    tips_text = '\n'.join(sections["Improvement Tips"])
+                    st.markdown(f'<div class="tip-box">{tips_text}</div>', unsafe_allow_html=True)
+                
+                if sections["Detailed Feedback"]:
+                    st.markdown("### 📋 Detailed Feedback")
+                    st.write('\n'.join(sections["Detailed Feedback"]))
+                
+                # Show full AI response in expander
+                with st.expander("View Full AI Analysis"):
+                    st.markdown(result)
+                
+        # --- Question Answering Section ---
+        st.markdown("---")
+        st.markdown('<p class="subheader">❓ Ask About Your Report</p>', unsafe_allow_html=True)
+        
+        col1, col2 = st.columns([3, 1])
+        with col1:
+            user_question = st.text_input("Ask a question about your lab report", 
+                                         placeholder="e.g., How can I improve my hypothesis?")
+        with col2:
+            st.markdown("<div style='height: 28px;'></div>", unsafe_allow_html=True)
+            ask_button = st.button("🔍 Ask Question", use_container_width=True)
+        
+        if (ask_button or user_question) and user_question.strip():
+            with st.spinner("Thinking..."):
+                followup_prompt = f"""Lab Report:
+                {lab_text}
+                
+                Question: {user_question}
+                
+                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)
             
-            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)
+            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("---")
+        st.markdown('<p class="subheader">📝 Sample Lab Report</p>', unsafe_allow_html=True)
+        st.markdown("""
+        **Title:** Effect of Temperature on Enzyme Activity  
+        **Objective:** To investigate how temperature affects catalase enzyme activity  
+        **Hypothesis:** Enzyme activity will increase with temperature up to 37°C, then decrease  
+        **Materials:** Test tubes, hydrogen peroxide, liver extract, thermometer  
+        **Procedure:**  
+        1. Prepare test tubes at 5 different temperatures  
+        2. Add equal amounts of hydrogen peroxide and liver extract  
+        3. Measure oxygen production  
+        **Observations:** More bubbles at 37°C compared to lower or higher temperatures  
+        **Conclusion:** Enzyme activity peaks at body temperature  
+        """)
         
-        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("---")
-    st.markdown('<p class="subheader">📝 Sample Lab Report</p>', unsafe_allow_html=True)
-    st.markdown("""
-    **Title:** Effect of Temperature on Enzyme Activity  
-    **Objective:** To investigate how temperature affects catalase enzyme activity  
-    **Hypothesis:** Enzyme activity will increase with temperature up to 37°C, then decrease  
-    **Materials:** Test tubes, hydrogen peroxide, liver extract, thermometer  
-    **Procedure:**  
-    1. Prepare test tubes at 5 different temperatures  
-    2. Add equal amounts of hydrogen peroxide and liver extract  
-    3. Measure oxygen production  
-    **Observations:** More bubbles at 37°C compared to lower or higher temperatures  
-    **Conclusion:** Enzyme activity peaks at body temperature  
-    """)
-    
-    st.info("👆 Upload your own lab report to get a personalized analysis!")
+        st.info("👆 Upload your own lab report to get a personalized analysis!")
+
+# --- Feedback Section ---
+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.")