Update app.py

app.py

@@ -179,9 +179,12 @@ def bot(history, cross_encoder):
             yield history, prompt_html
         #store_message(db,history[-1][0],history[-1][1],cross_encoder)
 
+def system_instructions(question_difficulty, topic,document_summaries):
+    return f"""<s> [INST] Your are a great teacher and your task is to create 10 questions with 4 choices with a {question_difficulty} difficulty  about topic request " {topic} " only from the below given documents, {'\n'.join(document_summaries)} then create an answers. Index in JSON format, the questions as "Q#":"" to "Q#":"", the four choices as "Q#:C1":"" to "Q#:C4":"", and the answers as "A#":"Q#:C#" to "A#":"Q#:C#". [/INST]"""
+
 
 #with gr.Blocks(theme='Insuz/SimpleIndigo') as demo:
-with gr.Blocks(theme='NoCrypt/miku') as demo:
+with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
     with gr.Row():
         with gr.Column(scale=10):
             # gr.Markdown(
@@ -264,5 +267,153 @@ with gr.Blocks(theme='NoCrypt/miku') as demo:
     # Examples
     gr.Examples(examples, txt)
 
+
+RAG_db=gr.State()
+
+with gr.Blocks(title="Quiz Maker", theme=gr.themes.Default(primary_hue="green", secondary_hue="green"), css="style.css") as QUIZBOT:
+    def load_model():
+        RAG= RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")
+        RAG_db.value=RAG.from_index('.ragatouille/colbert/indexes/cbseclass10index')
+        return 'Ready to Go!!'
+    with gr.Columns(scale=4):
+        gr.HTML("""
+    <center>
+      <h1><span style="color: purple;">AI NANBAN</span> - CBSE Class Quiz Maker</h1>
+      <h2>AI-powered Learning Game</h2>
+      <i>⚠️ Students create quiz from any topic /CBSE Chapter ! ⚠️</i>
+    </center>
+    """)
+        #gr.Warning('Retrieving using ColBERT.. First time query will take a minute for model to load..pls wait')
+    with gr.Columns(scale=2):
+        load_btn = gr.Button("Click to Load!🚀")
+        load_text=gr.Textbox()
+        load_btn.click(load_model,[],load_text)
+        
+   
+    topic = gr.Textbox(label="Enter the Topic for Quiz", placeholder="Write any topic from CBSE notes")
+
+    with gr.Row():
+        radio = gr.Radio(
+            ["easy", "average", "hard"], label="How difficult should the quiz be?"
+        )
+
+
+    generate_quiz_btn = gr.Button("Generate Quiz!🚀")
+
+    question_radios = [gr.Radio(visible=False), gr.Radio(visible=False), gr.Radio(
+        visible=False), gr.Radio(visible=False), gr.Radio(visible=False), gr.Radio(visible=False), gr.Radio(visible=False), gr.Radio(
+        visible=False), gr.Radio(visible=False), gr.Radio(visible=False)]
+
+    print(question_radios)
+
+    @spaces.GPU
+    @generate_quiz_btn.click(inputs=[radio, topic], outputs=question_radios, api_name="generate_quiz")
+    def generate_quiz(question_difficulty, topic, user_prompt):
+        top_k_rank=10
+        RAG_db_=RAG_db.value
+        documents_full=RAG_db_.search(topic,k=top_k_rank)
+    
+        documents=[item['content'] for item in documents_full]
+        document_summaries = [f"[DOCUMENT {i+1}]: {summary}" for i, summary in enumerate(documents)]
+        formatted_prompt = system_instructions(
+            question_difficulty, topic,document_summaries)
+        print(formatted_prompt)
+        pre_prompt = [
+            {"role": "system", "content": formatted_prompt}
+        ]
+
+        generate_kwargs = dict(
+            temperature=0.2,
+            max_new_tokens=4000,
+            top_p=0.95,
+            repetition_penalty=1.0,
+            do_sample=True,
+            seed=42,
+        )
+        while True:
+            try:
+                response = client.text_generation(
+                    formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False,
+                )
+                output_json = json.loads(f"{response}")
+                break
+            except Exception as e:
+                print(f"Exception occurred: {e}")
+                print("Trying again...please wait")
+                continue
+        response = client.text_generation(
+            formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False,
+        )
+
+        print(response)
+        print('output json', output_json)
+
+        global quiz_data
+
+        quiz_data = output_json
+
+        question_radio_list = []
+
+        for question_num in range(1, 11):
+            question_key = f"Q{question_num}"
+            answer_key = f"A{question_num}"
+
+            question = quiz_data.get(question_key)
+            answer = quiz_data.get(quiz_data.get(answer_key))
+
+            if not question or not answer:
+                continue
+
+            choice_keys = [f"{question_key}:C{i}" for i in range(1, 5)]
+            choice_list = []
+            for choice_key in choice_keys:
+                choice = quiz_data.get(choice_key, "Choice not found")
+                choice_list.append(f"{choice}")
+
+            radio = gr.Radio(choices=choice_list, label=question,
+                             visible=True, interactive=True)
+
+            question_radio_list.append(radio)
+
+        print('Question radio list ' , question_radio_list)
+
+        return question_radio_list
+
+    check_button = gr.Button("Check Score")
+
+    score_textbox = gr.Markdown()
+
+    @check_button.click(inputs=question_radios, outputs=score_textbox)
+    def compare_answers(*user_answers):
+        user_anwser_list = []
+        user_anwser_list = user_answers
+
+        answers_list = []
+
+        for question_num in range(1, 20):
+            answer_key = f"A{question_num}"
+            answer = quiz_data.get(quiz_data.get(answer_key))
+            if not answer:
+                break
+            answers_list.append(answer)
+
+        score = 0
+
+        for item in user_anwser_list:
+            if item in answers_list:
+                score += 1
+        if score>5:
+             message = f"### Good ! You got {score} over 10!"
+        elif score>7:
+             message = f"### Excellent ! You got {score} over 10!"
+        else:
+             message = f"### You got {score} over 10! Dont worry . You can prepare well and try better next time !"
+
+        return message
+
+
+
+demo = gr.TabbedInterface([CHATBOT,QUIZBOT], ["AI ChatBot", "AI Nanban-Quizbot"])
+
 demo.queue()
 demo.launch(debug=True)