Update app.py

app.py

@@ -6,31 +6,36 @@ import os
 
 # --- CONFIGURATION ---
 EXCEL_CSV_URL = "https://docs.google.com/spreadsheets/d/1b_gBz-_TZZNJLIM5sYyYJQfBVNua7dFf/export?format=csv"
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Make sure it's set in Hugging Face or locally
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Must be set in Hugging Face or .env
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 LLAMA3_MODEL = "llama3-8b-8192"
 
 # --- FUNCTIONS ---
 
 def load_excel_data():
-    response = requests.get(EXCEL_CSV_URL)
-    response.raise_for_status()
-    return pd.read_csv(BytesIO(response.content))
+    try:
+        response = requests.get(EXCEL_CSV_URL)
+        response.raise_for_status()
+        df = pd.read_csv(BytesIO(response.content))
+        st.write("✅ **Columns Detected in Sheet:**", list(df.columns))  # Help with debugging
+        return df
+    except Exception as e:
+        st.error(f"Error loading sheet: {e}")
+        return pd.DataFrame()
 
 def search_student_by_roll(data, roll_number):
-    df = data.copy()
-    results = df[df['RollNumber'].astype(str).str.lower() == roll_number.lower()]
-    return results
+    roll_column = [col for col in data.columns if 'roll' in col.lower()][0]  # Find roll column
+    result = data[data[roll_column].astype(str).str.lower() == roll_number.lower()]
+    return result
 
 def generate_summary_from_data(student_row):
     content = student_row.to_dict(orient="records")[0]
     prompt = (
-        f"Create a detailed student result summary using this data:\n\n"
+        f"Create a well-formatted result sheet using this student data:\n\n"
         f"{content}\n\n"
-        f"Format the result similar to a formal report card or academic result sheet. "
-        f"Include name, class, roll number, subjects with marks, total, percentage, and performance remarks."
+        f"Include Name, Roll Number, Class, Subject-wise Marks, Total, Percentage, and a Performance Remark."
     )
-    
+
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
@@ -38,10 +43,11 @@ def generate_summary_from_data(student_row):
     data = {
         "model": LLAMA3_MODEL,
         "messages": [
-            {"role": "system", "content": "You are a result sheet generator for a school."},
+            {"role": "system", "content": "You are a student result generator for a school."},
             {"role": "user", "content": prompt}
         ]
     }
+
     response = requests.post(GROQ_URL, headers=headers, json=data)
     response.raise_for_status()
     return response.json()["choices"][0]["message"]["content"]
@@ -57,12 +63,14 @@ def main():
     roll_number = st.text_input("Roll Number", placeholder="e.g., k0000")
 
     if st.button("Generate Result"):
-        with st.spinner("Fetching and generating result..."):
+        with st.spinner("📥 Loading data and generating result..."):
+            data = load_excel_data()
+            if data.empty:
+                return
             try:
-                data = load_excel_data()
                 result = search_student_by_roll(data, roll_number)
                 if not result.empty:
-                    st.success("✅ Student Found! Generating result summary...")
+                    st.success("✅ Student found. Generating summary...")
                     summary = generate_summary_from_data(result)
                     st.markdown("### 📝 Result Summary:")
                     st.markdown(summary)