Update app.py

app.py

@@ -273,165 +273,294 @@ with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
     gr.Examples(examples, txt)
 
 
-RAG_db=gr.State()
+
+RAG_db = gr.State()
+
+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!!'
+
+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)
+            
+            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}")
+            
+            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 = [quiz_data.get(choice_key, "Choice not found") for choice_key in choice_keys]
+                
+                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:
+                count += 1
+                continue
+        except Exception as e:
+            count += 1
+            if count == 3:
+                return ['Sorry. Pls try with another topic!'] + [gr.Radio(visible=False) for _ in range(10)]
+            continue
+    
+    return ['Quiz Generated!'] + question_radio_list
+
+def compare_answers(*user_answers):
+    user_answer_list = user_answers
+    answers_list = [quiz_data.get(quiz_data.get(f"A{question_num}")) for question_num in range(1, 11)]
+    
+    score = sum(1 for answer in user_answer_list if answer in answers_list)
+    
+    if score > 7:
+        message = f"### Excellent! You got {score} out of 10!"
+    elif score > 5:
+        message = f"### Good! You got {score} out of 10!"
+    else:
+        message = f"### You got {score} out of 10! Don’t worry, you can prepare well and try better next time!"
+    
+    return message
 
 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.Column(scale=4):
         gr.HTML("""
-    <center>
-      <h1><span style="color: purple;">ADWITIYA</span> Customs Manual  Quizbot</h1>
-      <h2>Generative AI-powered Capacity building for Training Officers</h2>
-      <i>⚠️ NACIN Faculties create quiz from any topic dynamically for classroom evaluation after their sessions ! ⚠️</i>
-    </center>
-    """)
-        #gr.Warning('Retrieving using ColBERT.. First time query will take a minute for model to load..pls wait')
+        <center>
+            <h1><span style="color: purple;">ADWITIYA</span> Customs Manual Quizbot</h1>
+            <h2>Generative AI-powered Capacity building for Training Officers</h2>
+            <i>⚠️ NACIN Faculties create quiz from any topic dynamically for classroom evaluation after their sessions! ⚠️</i>
+        </center>
+        """)
+    
     with gr.Column(scale=2):
         load_btn = gr.Button("Click to Load!🚀")
-        load_text=gr.Textbox()
-        load_btn.click(load_model,[],load_text)
-        
-   
+        load_text = gr.Textbox()
+        load_btn.click(fn=load_model, inputs=[], outputs=load_text)
+    
     topic = gr.Textbox(label="Enter the Topic for Quiz", placeholder="Write any topic/details from Customs Manual")
 
     with gr.Row():
-        radio = gr.Radio(
-            ["easy", "average", "hard"], label="How difficult should the quiz be?"
-        )
+        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) for _ in range(10)]
+    
+    generate_quiz_btn.click(
+        fn=generate_quiz, 
+        inputs=[radio, topic], 
+        outputs=[quiz_msg] + question_radios
+    )
+    
+    check_button = gr.Button("Check Score")
+    score_textbox = gr.Markdown()
+    
+    check_button.click(
+        fn=compare_answers,
+        inputs=question_radios,
+        outputs=score_textbox
+    )
 
+demo = gr.TabbedInterface([CHATBOT, QUIZBOT], ["AI ChatBot", "AI Quizbot"])
+demo.queue()
+demo.launch(debug=True)
 
-    generate_quiz_btn = gr.Button("Generate Quiz!🚀")
-    quiz_msg=gr.Textbox()
+# 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.Column(scale=4):
+#         gr.HTML("""
+#     <center>
+#       <h1><span style="color: purple;">ADWITIYA</span> Customs Manual  Quizbot</h1>
+#       <h2>Generative AI-powered Capacity building for Training Officers</h2>
+#       <i>⚠️ NACIN Faculties create quiz from any topic dynamically for classroom evaluation after their sessions ! ⚠️</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/details from Customs Manual")
 
-    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)]
+#     with gr.Row():
+#         radio = gr.Radio(
+#             ["easy", "average", "hard"], label="How difficult should the quiz be?"
+#         )
 
-    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_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}")
+#         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)
+#                 print(response)
+#                 print('output json', output_json)
         
-                global quiz_data
+#                 global quiz_data
         
-                quiz_data = output_json
+#                 quiz_data = output_json
         
                 
         
-                for question_num in range(1, 11):
-                    question_key = f"Q{question_num}"
-                    answer_key = f"A{question_num}"
+#                 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))
+#                     question = quiz_data.get(question_key)
+#                     answer = quiz_data.get(quiz_data.get(answer_key))
         
-                    if not question or not answer:
-                        continue
+#                     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}")
+#                     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)
+#                     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
+#                     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)
+#         print('Question radio list ' , question_radio_list)
 
-        return ['Quiz Generated!']+ question_radio_list
+#         return ['Quiz Generated!']+ question_radio_list
 
-    check_button = gr.Button("Check Score")
+#     check_button = gr.Button("Check Score")
 
-    score_textbox = gr.Markdown()
+#     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
+#     @check_button.click(inputs=question_radios, outputs=score_textbox)
+#     def compare_answers(*user_answers):
+#         user_anwser_list = []
+#         user_anwser_list = user_answers
 
-        answers_list = []
+#         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)
+#         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
+#         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 !"
+#         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
+#         return message
 
 
 
-demo = gr.TabbedInterface([CHATBOT,QUIZBOT], ["AI ChatBot", "AI Quizbot"])
+# demo = gr.TabbedInterface([CHATBOT,QUIZBOT], ["AI ChatBot", "AI Quizbot"])
 
-demo.queue()
-demo.launch(debug=True)
+# demo.queue()
+# demo.launch(debug=True)