Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import google.generativeai as genai
 from dotenv import load_dotenv
 import os
+import json
 
 # Load environment variables
 load_dotenv()
@@ -10,59 +11,85 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 model = genai.GenerativeModel('gemini-pro')
 
+# Load questionnaire from JSON
+def load_questionnaire():
+    with open('assets/questionnaire.json', 'r') as f:
+        return json.load(f)
+
+# Render questions based on type
+def render_question(question):
+    question_type = question.get("type")
+    if question_type == "slider":
+        return st.slider(
+            question["question"],
+            min_value=question.get("min", 1),
+            max_value=question.get("max", 10),
+            value=question.get("default", 5)
+        )
+    elif question_type == "select_slider":
+        return st.select_slider(
+            question["question"],
+            options=question["options"],
+            value=question.get("default")
+        )
+    elif question_type == "radio":
+        return st.radio(
+            question["question"],
+            options=question["options"]
+        )
+    elif question_type == "select":
+        return st.selectbox(
+            question["question"],
+            options=question["options"]
+        )
+    elif question_type == "multiselect":
+        return st.multiselect(
+            question["question"],
+            options=question["options"]
+        )
+    elif question_type == "number":
+        return st.number_input(
+            question["question"],
+            min_value=question.get("min", 0),
+            max_value=question.get("max", 24),
+            value=question.get("default", 4)
+        )
+    else:
+        st.warning(f"Unsupported question type: {question_type}")
+        return None
+
+# Main function
 def main():
     st.title("JEE SOCA Analysis System 🚀")
     st.subheader("AI-Powered Skill Assessment for JEE Aspirants")
 
+    # Load questionnaire
+    questionnaire = load_questionnaire()
+
+    # Collect responses
+    responses = {}
     with st.form("student_form"):
         st.header("Student Questionnaire")
         
-        # Subject Knowledge
-        st.subheader("📚 Subject Knowledge")
-        math_score = st.slider("Mathematics Proficiency (1-10)", 1, 10)
-        physics_score = st.slider("Physics Proficiency (1-10)", 1, 10)
-        chemistry_score = st.slider("Chemistry Proficiency (1-10)", 1, 10)
-        
-        # Study Habits
-        st.subheader("⏰ Study Habits")
-        study_hours = st.number_input("Daily Study Hours", min_value=0, max_value=16)
-        time_management = st.select_slider("Time Management Skills", 
-                                         options=["Poor", "Average", "Good", "Excellent"])
-        
-        # Problem Solving
-        st.subheader("🔍 Problem Solving")
-        problem_approach = st.radio("Preferred Problem Solving Approach",
-                                  ("Conceptual Understanding", 
-                                   "Formula Memorization", 
-                                   "Practice Problems"))
+        # Render questions from JSON
+        for section in questionnaire["questionnaire"]:
+            st.subheader(f"📚 {section['section']}")
+            for question in section["questions"]:
+                response = render_question(question)
+                if response is not None:
+                    responses[question["question"]] = response
         
-        # Stress Management
-        st.subheader("🧘 Stress Management")
-        stress_level = st.select_slider("Stress Level", 
-                                      options=["Very Low", "Low", "Moderate", "High", "Very High"])
-        
-        # Submission
+        # Submit button
         submitted = st.form_submit_button("Generate SOCA Analysis")
         
         if submitted:
             with st.spinner("Analyzing responses..."):
-                # Prepare prompt
-                prompt = f"""Analyze this JEE student's profile and create a SOCA analysis:
-                Subject Scores:
-                - Mathematics: {math_score}/10
-                - Physics: {physics_score}/10
-                - Chemistry: {chemistry_score}/10
-                
-                Study Habits:
-                - Daily Study Hours: {study_hours}
-                - Time Management: {time_management}
-                
-                Problem Solving:
-                - Approach: {problem_approach}
-                
-                Stress Management:
-                - Stress Level: {stress_level}
+                # Prepare prompt for Gemini
+                prompt = "Analyze this JEE student's profile and create a SOCA analysis:\n\n"
+                for question, response in responses.items():
+                    prompt += f"- {question}: {response}\n"
                 
+                prompt += """
                 Provide the analysis in this format:
                 **Strengths:** [Identify 3 key strengths]
                 **Opportunities:** [Suggest 3 improvement areas]
@@ -71,11 +98,13 @@ def main():
                 """
                 
                 # Get Gemini response
-                response = model.generate_content(prompt)
-                
-                # Display analysis
-                st.subheader("SOCA Analysis Report")
-                st.markdown(response.text)
+                try:
+                    response = model.generate_content(prompt)
+                    st.subheader("SOCA Analysis Report")
+                    st.markdown(response.text)
+                except Exception as e:
+                    st.error(f"An error occurred while generating the analysis: {e}")
 
+# Run the app
 if __name__ == "__main__":
     main()