Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
-import streamlit as st
 from dotenv import load_dotenv
+import streamlit as st
 from groq import Groq
 
 # Load environment variables from .env file
@@ -12,7 +12,7 @@ api_key = os.getenv("GROQ_API_KEY")
 # Initialize Groq client with the API key
 client = Groq(api_key=api_key)
 
-# Function to generate MCQs based on the input text
+# Function to generate MCQs from the user text
 def generate_mcqs_from_text(user_text):
     prompt = f"""
     You are a 35-year experienced educator specializing in crafting challenging and insightful MCQs. 
@@ -36,13 +36,13 @@ def generate_mcqs_from_text(user_text):
     """
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
-        model="Llama-3.3-70b",  # Replace with a valid Groq model
+        model="gemma2-9b-it",  # Replace with a valid Groq model
     )
     response = chat_completion.choices[0].message.content
     print("API Response:\n", response)  # Debugging: Inspect the raw response
     return response
 
-# Function to parse the MCQs and correct answers from the API response
+# Function to parse the MCQs and correct answers
 def parse_mcqs_and_answers(mcqs_and_answers):
     try:
         # Try splitting on "Correct Answers:"
@@ -64,47 +64,62 @@ def parse_mcqs_and_answers(mcqs_and_answers):
         st.write("Raw response:", mcqs_and_answers)  # Display raw output for troubleshooting
         return None, None
 
-# Streamlit app
-def main():
-    st.title("MCQ Generator and Evaluator")
+# Function to evaluate user answers
+def evaluate_answers(mcqs, user_answers):
+    # Here we assume the user_answers is a dictionary of answers for each question
+    prompt = f"""
+    Here are the user's answers to the following multiple-choice questions. 
+    Please evaluate them as an experienced teacher, providing feedback on correctness, 
+    and explain why each answer is right or wrong.
+
+    MCQs: {mcqs}
+
+    User Answers: {user_answers}
+    """
     
-    # Input text from the user
-    user_text = st.text_area("Enter the text for MCQ generation:", height=150)
+    # Make the API call to evaluate the answers
+    chat_completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": prompt}],
+        model="gemma2-9b-it",  # Replace with a valid Groq model
+    )
     
-    if st.button("Generate MCQs"):
-        if user_text:
-            # Generate MCQs from the input text
-            mcqs_and_answers = generate_mcqs_from_text(user_text)
-            
-            # Parse the MCQs and correct answers
-            mcqs, correct_answers = parse_mcqs_and_answers(mcqs_and_answers)
-            
-            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.")
+    # Return the evaluation result
+    return chat_completion.choices[0].message.content
+
+# Streamlit UI
+st.title("MCQ Generator and Evaluator")
+
+# User input
+user_text = st.text_area("Enter text to generate MCQs:")
+
+if st.button("Generate MCQs"):
+    # Generate MCQs from user-provided text
+    mcqs_and_answers = generate_mcqs_from_text(user_text)
+    
+    # Parse the MCQs and correct answers
+    mcqs, correct_answers = parse_mcqs_and_answers(mcqs_and_answers)
     
-if __name__ == "__main__":
-    main()
+    if mcqs and correct_answers:
+        # Display the first MCQ question
+        questions = mcqs.splitlines()
+        question = questions[0]
+        
+        # Show question options
+        st.write(f"Question: {question}")
+        
+        options = ["A", "B", "C", "D"]  # Example options, you may adjust according to your response
+        user_answer = st.radio("Choose your answer:", options)
+        
+        if st.button("Submit Answer"):
+            # Evaluate the answer
+            result = evaluate_answers(mcqs, {question: user_answer})
+            st.write(result)
+            
+            # Show next question if available
+            if len(questions) > 1:
+                next_question = questions[1]
+                st.write(f"Next Question: {next_question}")
+            else:
+                st.write("You have completed all the questions!")
+    else:
+        st.error("Could not parse MCQs or answers.")