Update app.py

app.py

@@ -118,7 +118,7 @@ def bot(history, api_kind):
         print('An unexpected error occurred during generation:', str(e))
         yield f"An unexpected error occurred during generation: {str(e)}"
 
-with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
+with gr.Blocks(theme='WeixuanYuan/Soft_dark') as CHATBOT:
     # Beautiful heading with logo
     gr.HTML(value="""
     <div style="display: flex; align-items: center; justify-content: space-between;">
@@ -173,6 +173,173 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
 
     # Examples
     gr.Examples(examples, txt)
+    
+    # QUIZBOT CODE
+    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 system_instructions(question_difficulty, topic,documents_str):
+        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, {documents_str} 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]"""
+
+    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.Column(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.Column(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!🚀")
+    quiz_msg=gr.Textbox()
+
+    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=[quiz_msg]+question_radios, api_name="generate_quiz")
+    def generate_quiz(question_difficulty, topic):
+        top_k_rank=10
+        RAG_db_=RAG_db.value
+        documents_full=RAG_db_.search(topic,k=top_k_rank)
+    
+        
+
+        generate_kwargs = dict(
+            temperature=0.2,
+            max_new_tokens=4000,
+            top_p=0.95,
+            repetition_penalty=1.0,
+            do_sample=True,
+            seed=42,
+        )
+        question_radio_list = []
+        count=0
+        while count<=3:
+            try:
+                documents=[item['content'] for item in documents_full]
+                document_summaries = [f"[DOCUMENT {i+1}]: {summary}{count}" for i, summary in enumerate(documents)]
+                documents_str='\n'.join(document_summaries)
+                formatted_prompt = system_instructions(
+                    question_difficulty, topic,documents_str)
+                print(formatted_prompt)
+                pre_prompt = [
+                    {"role": "system", "content": formatted_prompt}
+                ]
+                response = client.text_generation(
+                    formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False,
+                )
+                output_json = json.loads(f"{response}")
+                
+        
+                print(response)
+                print('output json', output_json)
+        
+                global quiz_data
+        
+                quiz_data = output_json
+        
+                
+        
+                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)
+                if len(question_radio_list)==10:
+                    break
+                else:
+                    print('10 questions not generated . So trying again!')
+                    count+=1
+                    continue
+            except Exception as e:
+                count+=1
+                print(f"Exception occurred: {e}")
+                if count==3:
+                    print('Retry exhausted')
+                    gr.Warning('Sorry. Pls try with another topic !')
+                else:
+                    print(f"Trying again..{count} time...please wait")
+                    continue
+
+        print('Question radio list ' , question_radio_list)
+
+        return ['Quiz Generated!']+ 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)