Update app.py

app.py

@@ -187,6 +187,98 @@ def bot(history, cross_encoder):
 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]"""
 
+RAG_db = gr.State()
+
+def load_model():
+    try:
+        # Initialize the model
+        RAG = RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")
+        # Load the RAG database
+        RAG_db.value = RAG.from_index('.ragatouille/colbert/indexes/cbseclass10index')
+        return 'Ready to Go!!'
+    except Exception as e:
+        return f"Error loading model: {e}"
+    
+
+def generate_quiz(question_difficulty, topic):
+    if not topic.strip():
+        return ['Please enter a valid topic.'] + [gr.Radio(visible=False) for _ in range(10)]
+  
+    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(theme='Insuz/SimpleIndigo') as demo:
 with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
@@ -274,89 +366,6 @@ with gr.Blocks(theme='NoCrypt/miku') as CHATBOT:
 
 
 
-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:
     with gr.Column(scale=4):