Update app.py

app.py

@@ -1,64 +1,43 @@
 import streamlit as st
 import requests
 
-# Hugging Face API setup
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
-headers = {"Authorization": "Bearer YOUR_HF_API_KEY"}
+# Open-source LLM endpoint (replace if needed)
+API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
+
+# If you have a Hugging Face token, include it here. If not, this model works without one.
+headers = {"Authorization": ""}
 
 def query_llm(prompt):
     payload = {"inputs": prompt}
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()[0]['generated_text']
 
-# Static templates for known experiments
-experiment_templates = {
-    "Vinegar and Baking Soda": {
-        "goal": "Observe a chemical reaction and gas formation.",
-        "materials": "Vinegar, Baking Soda, Container",
-        "default_hypothesis": "Mixing will cause bubbles or fizz due to gas release.",
-        "result": "Bubbles and fizz from CO2 gas produced in an acid-base reaction.",
-        "explanation": "Acetic acid in vinegar reacts with sodium bicarbonate to form carbon dioxide gas."
-    },
-    "Lemon Battery": {
-        "goal": "Generate electricity using a lemon as a battery.",
-        "materials": "Lemon, Copper coin, Zinc nail, Wires, LED",
-        "default_hypothesis": "The lemon will generate voltage to light up a small LED.",
-        "result": "LED glows slightly due to electron flow.",
-        "explanation": "The citric acid acts as electrolyte between the copper and zinc electrodes."
-    }
-}
+    try:
+        output = response.json()
+        if isinstance(output, list) and 'generated_text' in output[0]:
+            return output[0]['generated_text']
+        elif isinstance(output, dict) and "error" in output:
+            return f"❌ API Error: {output['error']}"
+        else:
+            return "❌ Unexpected API response format."
+    except Exception as e:
+        return f"❌ Failed to parse response: {str(e)}"
 
 # Streamlit UI
-st.set_page_config(page_title="Science Lab Assistant", layout="centered")
+st.set_page_config(page_title="Science Lab Assistant", page_icon="🧪")
 st.title("🧪 Science Lab Assistant")
-
-# Select known or custom experiment
-exp_option = st.selectbox("Choose an experiment", ["Custom"] + list(experiment_templates.keys()))
-
-if exp_option != "Custom":
-    data = experiment_templates[exp_option]
-    st.subheader("📋 Experiment Summary")
-    st.write(f"**Goal:** {data['goal']}")
-    st.write(f"**Materials:** {data['materials']}")
-    st.write(f"**Suggested Hypothesis:** {data['default_hypothesis']}")
-
-    if st.button("🔍 Show Result & Explanation"):
-        st.success(f"**Result:** {data['result']}")
-        st.info(f"**Why?:** {data['explanation']}")
-
-else:
-    st.subheader("🔬 Describe your experiment")
-    experiment_name = st.text_input("Name or short description of your experiment")
-    goal = st.text_area("What is the goal of your experiment?")
-    materials = st.text_area("List the materials involved")
-
-    if st.button("🧠 Generate Hypothesis and Expected Result"):
-        prompt = f"""I am conducting a school science experiment. 
-        Experiment: {experiment_name}
-        Goal: {goal}
-        Materials: {materials}
-        Please suggest a hypothesis and likely result with a brief scientific explanation."""
-        result = query_llm(prompt)
-        st.markdown("### 🤖 Assistant's Suggestion")
-        st.write(result)
-
+st.subheader("Ask me about school science experiments!")
+
+with st.form("lab_form"):
+    experiment = st.text_input("🔍 Enter your experiment name or question:")
+    hypothesis = st.text_area("💡 What is your hypothesis?")
+    submitted = st.form_submit_button("Explain Experiment")
+
+if submitted and experiment:
+    prompt = f"Explain the experiment '{experiment}' for a school science class. The student hypothesizes: '{hypothesis}'. Provide an explanation and possible result."
+    result = query_llm(prompt)
+    st.markdown("### 🧠 AI Explanation")
+    st.write(result)
+
+st.markdown("---")
+st.markdown("📚 Example: `Vinegar and baking soda`")
+st.markdown("💡 Hypothesis: `Combining them will create bubbles.`")