Update app.py

app.py

@@ -4,10 +4,8 @@ import streamlit as st
 from huggingface_hub import InferenceClient
 from dotenv import load_dotenv
 
-# Load environment variables for local development
+# Load environment variables
 load_dotenv()
-
-# Initialize the Hugging Face Inference client
 hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 client = InferenceClient(provider="auto", api_key=hf_token)
 
@@ -19,7 +17,7 @@ st.set_page_config(
 )
 st.title("🎓 Interview Preparation Chatbot")
 
-# Session state initialization
+# Initialize session state
 if "questions" not in st.session_state:
     st.session_state.questions = []
 if "topic" not in st.session_state:
@@ -31,13 +29,28 @@ if "correct_count" not in st.session_state:
 if "incorrect_count" not in st.session_state:
     st.session_state.incorrect_count = 0
 
-# Sidebar: Topic and Score display
+# Sidebar: Topic selection and scoring
 st.sidebar.header("Practice Topic")
 st.session_state.topic = st.sidebar.selectbox(
     "Select a topic:",
-    ["Machine Learning","Data Structures","Python","Generative AI","Computer Vision","Deep Learning"],
-    index=["Machine Learning","Data Structures","Python","Generative AI","Computer Vision","Deep Learning"].index(st.session_state.topic)
+    [
+        "Machine Learning",
+        "Data Structures",
+        "Python",
+        "Generative AI",
+        "Computer Vision",
+        "Deep Learning",
+    ],
+    index=[
+        "Machine Learning",
+        "Data Structures",
+        "Python",
+        "Generative AI",
+        "Computer Vision",
+        "Deep Learning",
+    ].index(st.session_state.topic)
 )
+
 st.sidebar.markdown("---")
 st.sidebar.header("Your Score")
 st.sidebar.markdown(f"**Total:** {len(st.session_state.questions)}")
@@ -45,30 +58,36 @@ st.sidebar.markdown(f"**Correct:** {st.session_state.correct_count}")
 st.sidebar.markdown(f"**Incorrect:** {st.session_state.incorrect_count}")
 st.sidebar.markdown(f"**Points:** {st.session_state.score}")
 
-# Function to fetch a new MCQ question via HF Inference API
+# Function to fetch an MCQ question
 
 def fetch_question(topic):
     prompt = {
         "role": "system",
         "content": (
-            "You are an expert interviewer. "
-            "Generate a multiple-choice question on the topic of {topic}. "
-            "Respond with a JSON object: {\"question\": str, \"options\": [str, ...], \"correct_index\": int}"
-        ).format(topic=topic)
+            f"You are an expert interviewer. Generate a multiple-choice question on the topic of {topic}. "
+            "Respond with a JSON object: {\"question\": str, \"options\": [str, ...], \"correct_index\": int}."
+        )
     }
-    response = client.chat.completions.create(
-        model="mistralai/Mistral-7B-Instruct-v0.1",
-        messages=[prompt]
-    )
-    content = response.choices[0].message["content"].strip()
     try:
+        response = client.chat.completions.create(
+            model="mistralai/Mistral-7B-Instruct-v0.1",
+            messages=[prompt]
+        )
+        content = response.choices[0].message["content"].strip()
         data = json.loads(content)
-        return data
-    except json.JSONDecodeError:
-        st.error("Failed to parse question. Please try again.")
+    except Exception as e:
+        st.error(f"Failed to fetch or parse question: {e}")
+        return None
+    # Validate structure
+    question = data.get("question")
+    options = data.get("options")
+    correct_index = data.get("correct_index")
+    if not question or not isinstance(options, list) or correct_index is None:
+        st.error(f"Invalid question structure. Response: {content}")
         return None
+    return {"question": question, "options": options, "correct_index": correct_index}
 
-# Buttons for initial and next questions
+# Buttons to get or advance questions
 if not st.session_state.questions:
     if st.button("Get Question"):
         q = fetch_question(st.session_state.topic)
@@ -97,19 +116,19 @@ for idx, q in enumerate(st.session_state.questions):
     )
     st.session_state.questions[idx]["selected"] = sel
 
-    # Submit button
+    # Submit answer
     if not q["submitted"]:
         if st.button("Submit Answer", key=f"submit_{idx}"):
             st.session_state.questions[idx]["submitted"] = True
             if sel == q["correct_index"]:
-                st.success("Correct!")
+                st.success("Correct! +10 points")
                 st.session_state.score += 10
                 st.session_state.correct_count += 1
             else:
-                st.error(f"Incorrect! Correct answer: {opts[q['correct_index']]}.")
+                st.error(f"Incorrect! Correct: {opts[q['correct_index']]} (-10 points)")
                 st.session_state.score -= 10
                 st.session_state.incorrect_count += 1
 
-# Footer info
+# Footer
 st.markdown("---")
-st.markdown("*Points: +10 for correct, -10 for incorrect.*")
+st.markdown("*Correct: +10 pts | Incorrect: -10 pts*")