Update app.py

app.py

@@ -1,15 +1,18 @@
 import os
+import streamlit as st
 from dotenv import load_dotenv
 from groq import Groq
-import streamlit as st
 
-# Load environment variables
+# Load environment variables from .env file
 load_dotenv()
 
-# Initialize the Groq client with API key
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# Get the API key from the environment variables
+api_key = os.getenv("GROQ_API_KEY")
 
-# Function to generate MCQs from the user text
+# Initialize Groq client with the API key
+client = Groq(api_key=api_key)
+
+# Function to generate MCQs based on the input text
 def generate_mcqs_from_text(user_text):
     prompt = f"""
     You are a 35-year experienced educator specializing in crafting challenging and insightful MCQs. 
@@ -33,85 +36,75 @@ def generate_mcqs_from_text(user_text):
     """
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
-        model="gemma2-9b-it",  # Replace with a valid Groq model
+        model="Llama-3.3-70b",  # Replace with a valid Groq model
     )
-    return chat_completion.choices[0].message.content
+    response = chat_completion.choices[0].message.content
+    print("API Response:\n", response)  # Debugging: Inspect the raw response
+    return response
 
-# Function to parse MCQs and correct answers
+# Function to parse the MCQs and correct answers from the API response
 def parse_mcqs_and_answers(mcqs_and_answers):
     try:
-        mcqs, correct_answers_section = mcqs_and_answers.split("Correct Answers:")
-        correct_answers = {
-            f"Question {i+1}": answer.strip()
-            for i, answer in enumerate(correct_answers_section.splitlines() if correct_answers_section else [])
-        }
-        return mcqs.strip(), correct_answers
-    except ValueError:
-        st.error("The API response is not in the expected format. Please check the output or adjust the prompt.")
+        # Try splitting on "Correct Answers:"
+        parts = mcqs_and_answers.split("Correct Answers:")
+        if len(parts) != 2:
+            raise ValueError("Could not find 'Correct Answers:' in the response.")
+        
+        # Separate MCQs and correct answers
+        mcqs = parts[0].strip()
+        correct_answers = {}
+        for line in parts[1].strip().splitlines():
+            if line.startswith("Question"):
+                q, ans = line.split(":")
+                correct_answers[q.strip()] = ans.strip()
+        
+        return mcqs, correct_answers
+    except Exception as e:
+        st.error(f"Failed to parse response: {e}")
+        st.write("Raw response:", mcqs_and_answers)  # Display raw output for troubleshooting
         return None, None
 
 # Streamlit app
-st.title("Experienced Teacher's MCQ Generator & Interactive Evaluator")
-
-# Sidebar for text input
-user_text = st.sidebar.text_area("Enter the text for generating MCQs:")
-
-# Generate MCQs and answers
-if st.sidebar.button("Generate MCQs"):
-    if user_text.strip():
-        with st.spinner("Generating MCQs..."):
+def main():
+    st.title("MCQ Generator and Evaluator")
+    
+    # Input text from the user
+    user_text = st.text_area("Enter the text for MCQ generation:", height=150)
+    
+    if st.button("Generate MCQs"):
+        if user_text:
+            # Generate MCQs from the input text
             mcqs_and_answers = generate_mcqs_from_text(user_text)
-            mcqs, correct_answers = parse_mcqs_and_answers(mcqs_and_answers)
             
-            if mcqs and correct_answers:
-                st.session_state["mcqs"] = mcqs.split("\n\n")  # Split into individual questions
-                st.session_state["correct_answers"] = correct_answers
-                st.session_state["current_index"] = 0
-                st.session_state["score"] = 0
-                st.session_state["finished"] = False
-                st.success("MCQs generated successfully!")
-            else:
-                st.error("Failed to parse the MCQs or answers. Please check the output format.")
-    else:
-        st.error("Please enter text for generating MCQs.")
-
-# Display current MCQ
-if "mcqs" in st.session_state and not st.session_state.get("finished", False):
-    current_index = st.session_state["current_index"]
-    mcqs = st.session_state["mcqs"]
-    correct_answers = st.session_state["correct_answers"]
-
-    if current_index < len(mcqs):
-        st.subheader(f"Question {current_index + 1}")
-        current_question = mcqs[current_index]
-        st.text(current_question)
-
-        # User selects their answer
-        user_answer = st.radio(
-            "Choose your answer:",
-            ["A", "B", "C", "D"],
-            key=f"question_{current_index}",
-        )
-
-        # Submit button to evaluate the answer
-        if st.button("Submit Answer"):
-            correct_answer = correct_answers.get(f"Question {current_index + 1}", "Unknown")
-            if user_answer == correct_answer:
-                st.session_state["score"] += 1
-                st.success("Correct!")
-            else:
-                st.error(f"Incorrect. The correct answer is {correct_answer}.")
+            # Parse the MCQs and correct answers
+            mcqs, correct_answers = parse_mcqs_and_answers(mcqs_and_answers)
             
-            # Move to the next question
-            st.session_state["current_index"] += 1
-            if st.session_state["current_index"] >= len(mcqs):
-                st.session_state["finished"] = True
-    else:
-        st.session_state["finished"] = True
-
-# Final score display
-if st.session_state.get("finished", False):
-    st.subheader("Quiz Completed!")
-    total_questions = len(st.session_state["mcqs"])
-    score = st.session_state["score"]
-    st.success(f"Your score: {score}/{total_questions}")
+            if mcqs:
+                # Display MCQs one by one and evaluate answers
+                mcq_list = mcqs.splitlines()
+                correct_count = 0
+                for idx, mcq in enumerate(mcq_list):
+                    question = mcq.split("\n")[0]
+                    options = mcq.split("\n")[1:5]
+                    st.write(question)
+                    for option in options:
+                        st.write(option)
+                    
+                    user_answer = st.radio("Select your answer", options=["A", "B", "C", "D"], key=idx)
+                    
+                    if user_answer:
+                        # Evaluate the user's answer
+                        correct_answer = correct_answers.get(f"Question {idx+1}")
+                        if user_answer == correct_answer:
+                            st.success(f"Correct! The answer is {correct_answer}.")
+                            correct_count += 1
+                        else:
+                            st.error(f"Incorrect. The correct answer is {correct_answer}.")
+                
+                # Display final score
+                st.write(f"Your final score is {correct_count}/{len(mcq_list)}.")
+        else:
+            st.error("Please enter some text to generate MCQs.")
+    
+if __name__ == "__main__":
+    main()