Update QuizMaster.py

QuizMaster.py

@@ -1,57 +1,154 @@
-import streamlit as st
-from dotenv import load_dotenv
-import os
-import json
-import base64
-from langchain_google_genai import ChatGoogleGenerativeAI
-
-# Load environment variables
-load_dotenv()
-os.getenv("GOOGLE_API_KEY")
-
-# Constants for templates
-RESPONSE_JSON = {
-    # Example JSON response structure...
-}
-
-TEMPLATE = """
-Text:{text}
-You are an expert MCQ maker...
-"""
-
-TEMPLATE2 = """
-You are an expert English grammarian and writer...
-"""
-
-def show():    
-    st.header("Quiz Master :game_die:")
-    st.subheader("Generate your MCQ quiz instantly!")
-
-    TEXT = st.text_input("Quiz Topic:", key="input1")
-    NUMBER = st.text_input("Number of Questions:", key="input2")
-    SUBJECT = st.text_input("Subject Area:", key="input3")
-    TONE = st.text_input("Difficulty Level:", key="input4")
-
-    submit = st.button("Generate Quiz")
-
-    if submit and TEXT:
-        llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.9)
-        # Chain setup...
-        response = generate_evaluate_chain(
-            # Passing inputs...
-        )
-        # Process and display the generated quiz...
-        pdf = PDF()
-        # PDF generation code...
-        
-        with open(pdf_file_path, "rb") as pdf_file:
-            st.download_button(
-                label=f"Download {SUBJECT} Quiz PDF",
-                data=pdf_file,
-                file_name=f"{SUBJECT}_quiz.pdf",
-                mime="application/pdf",
-            )
-
-        # Display the PDF in the app
-        pdf_display = f'<iframe src="data:application/pdf;base64,{base64.b64encode(open(pdf_file_path, "rb").read()).decode()}" width="700" height="1000" type="application/pdf"></iframe>'
-        st.markdown(pdf_display, unsafe_allow_html=True)
+from fpdf import FPDF
+class PDF(FPDF):
+    def header(self):
+        self.set_font('Arial', 'B', 12)
+        self.cell(0, 10, 'MCQ Quiz', 0, 1, 'C')
+
+    def chapter_title(self, num, label):
+        self.set_font('Arial', '', 12)
+        self.cell(0, 10, 'Question %d: %s' % (num, label), 0, 1, 'L')
+        self.ln(5)
+
+    def chapter_body(self, body):
+        self.set_font('Arial', '', 12)
+        self.multi_cell(0, 10, body)
+        self.ln()
+
+    def add_question(self, num, question, options):
+        self.chapter_title(num, question)
+        for key, option in options.items():
+            self.chapter_body(f"{key}. {option}")
+        self.ln()
+
+    def add_answers_section(self, answers):
+        self.add_page()
+        self.set_font('Arial', 'B', 12)
+        self.cell(0, 10, 'Answers', 0, 1, 'C')
+        self.ln(10)
+        self.set_font('Arial', '', 12)
+        for num, answer in answers.items():
+            self.cell(0, 10, f"Question {num}: {answer}", 0, 1, 'L')
+
+
+
+import streamlit as st
+from dotenv import load_dotenv
+load_dotenv()
+import os
+import json
+import base64
+from langchain_google_genai import ChatGoogleGenerativeAI
+os.getenv("GOOGLE_API_KEY")
+RESPONSE_JSON = {
+    "1": {
+        "mcq": "multiple choice question",
+        "options": {
+            "a": "choice here",
+            "b": "choice here",
+            "c": "choice here",
+            "d": "choice here",
+        },
+        "correct": "correct answer",
+    },
+    "2": {
+        "mcq": "multiple choice question",
+        "options": {
+            "a": "choice here",
+            "b": "choice here",
+            "c": "choice here",
+            "d": "choice here",
+        },
+        "correct": "correct answer",
+    },
+    "3": {
+        "mcq": "multiple choice question",
+        "options": {
+            "a": "choice here",
+            "b": "choice here",
+            "c": "choice here",
+            "d": "choice here",
+        },
+        "correct": "correct answer",
+    },
+}
+TEMPLATE="""
+Text:{text}
+You are an expert MCQ maker. Given the above text, it is your job to \
+create a quiz  of {number} multiple choice questions for {subject} students in {tone} tone. 
+Make sure the questions are not repeated and check all the questions to be conforming the text as well.
+Make sure to format your response like  RESPONSE_JSON below  and use it as a guide. \
+Ensure to make {number} MCQs
+### RESPONSE_JSON
+{response_json}
+"""
+
+TEMPLATE2="""
+You are an expert english grammarian and writer. Given a Multiple Choice Quiz for {subject} students.\
+You need to evaluate the complexity of the question and give a complete analysis of the quiz. Only use at max 50 words for complexity analysis. 
+if the quiz is not at per with the cognitive and analytical abilities of the students,\
+update the quiz questions which needs to be changed and change the tone such that it perfectly fits the student abilities
+Quiz_MCQs:
+{quiz}
+Check from an expert English Writer of the above quiz:
+"""
+def show():    
+    st.header("MCQ_Generator")
+    TEXT=st.text_input("Input Prompt: ",key="input1")
+    NUMBER=st.text_input("Number of MCQs ",key="input2")
+    SUBJECT=st.text_input("Topic of MCQs ",key="input3")
+    TONE=st.text_input("Difficulty Level ",key="input4")
+
+    submit=st.button("Submit")
+
+    if submit and TEXT:
+        llm = ChatGoogleGenerativeAI(model="gemini-pro",temperature=0.9)
+        from langchain.prompts import PromptTemplate
+        from langchain.chains import LLMChain
+        from langchain.chains import SequentialChain
+        quiz_generation_prompt = PromptTemplate(
+            input_variables=["text", "number", "subject", "tone", "response_json"],
+            template=TEMPLATE
+        )
+        quiz_chain=LLMChain(llm=llm, prompt=quiz_generation_prompt, output_key="quiz", verbose=True)
+        quiz_evaluation_prompt=PromptTemplate(input_variables=["subject", "quiz"], template=TEMPLATE2)
+        review_chain=LLMChain(llm=llm, prompt=quiz_evaluation_prompt, output_key="review", verbose=True)
+        generate_evaluate_chain=SequentialChain(chains=[quiz_chain, review_chain], input_variables=["text", "number", "subject", "tone", "response_json"],
+                                        output_variables=["quiz", "review"], verbose=True,)
+        response=generate_evaluate_chain(
+            {
+                "text": TEXT,
+                "number": NUMBER,
+                "subject":SUBJECT,
+                "tone": TONE,
+                "response_json": json.dumps(RESPONSE_JSON)
+            }
+        )
+        quiz=response.get("quiz")
+        if '### RESPONSE_JSON\n' in quiz:
+            quiz = quiz.split('### RESPONSE_JSON\n')[1]
+            quiz = json.loads(quiz)
+        else:
+            quiz=json.loads(quiz)
+        pdf = PDF()
+        pdf.add_page()
+        pdf.set_title(SUBJECT+" Quiz")
+        answers = {}
+        for key, value in quiz.items():
+            question_num = int(key)
+            pdf.add_question(question_num, value["mcq"], value["options"])
+            answers[question_num] = value["correct"]
+        pdf.add_answers_section(answers)
+
+        pdf_file_path =SUBJECT+"_mcq.pdf"
+        pdf.output(pdf_file_path)
+        
+        with open(pdf_file_path, "rb") as pdf_file:
+            st.download_button(
+                label="Download "+SUBJECT+" Quiz PDF",
+                data=pdf_file,
+                file_name=SUBJECT+"_quiz.pdf",
+                mime="application/pdf",
+            )
+
+        pdf_display = f'<iframe src="data:application/pdf;base64,{base64.b64encode(open(pdf_file_path, "rb").read()).decode()}" width="700" height="1000" type="application/pdf"></iframe>'
+        st.markdown(pdf_display, unsafe_allow_html=True)