Update app.py

app.py

@@ -4,9 +4,9 @@ from langchain.prompts import ChatPromptTemplate
 from langchain_community.chat_models import ChatOpenAI
 from langchain.schema import StrOutputParser
 
-def create_the_question_prompt_template(num_questions, questions_type, context):
+def create_the_question_prompt_template(num_questions, questions_type, difficulty_level, context):
     """Create the prompt template for the questions generator app."""
-    template = f"""Create {num_questions} {questions_type} questions about the following concept/contents: {context}.
+    template = f"""Create {num_questions} {questions_type} questions keeping difficulty level as {difficulty_level} about the following concept/contents: {context}.
     The format of the question could be one of the following:
   - Multiple-choice:
     Questions:
@@ -103,7 +103,7 @@ def format_questions(questions):
     if current_question:
         formatted.append(current_question.strip())
     return("\n".join(formatted))
-    
+
 def format_answers(answers):
     """Format answers to display with consistent alignment."""
     lines = answers.split('\n')
@@ -115,16 +115,16 @@ def format_answers(answers):
 
     return "\n".join(formatted)
 
-def generate_questions(context, num_questions, questions_type):
+def generate_questions(context, num_questions, questions_type,difficulty_level):
     """Function to generate questions."""
     os.environ["OPENAI_API_KEY"] = "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"
 
     llm = ChatOpenAI(temperature=0.0)
-    prompt_template = create_the_question_prompt_template(num_questions, questions_type, context)
+    prompt_template = create_the_question_prompt_template(num_questions, questions_type, difficulty_level, context)
     chain = create_question_chain(prompt_template, llm)
 
     try:
-        question_response = chain.invoke({"questions_type": questions_type, "num_questions": num_questions, "context": context})
+        question_response = chain.invoke({"questions_type": questions_type, "num_questions": num_questions,"difficulty_level": difficulty_level, "context": context})
 
         # Log the entire response for debugging
         print("Question Response:", question_response)
@@ -141,6 +141,11 @@ with gr.Blocks() as demo:
     context_input = gr.Textbox(label="Context/Concept", placeholder="Enter the concept for the questions")
     num_questions_input = gr.Slider(label="Number of Questions", minimum=1, maximum=5, value=2, step=1)
     question_type_input = gr.Radio(label="Quiz Type", choices=["multiple-choice", "true-false", "open-ended"], value="multiple-choice")
+    dropdown = gr.Dropdown(
+        choices=["Easy", "Medium", "Hard"],  # Options in the selection box
+        label="Select difficulty level ",
+        value="Medium"  # Default value
+    )
     generate_btn = gr.Button("Generate Questions")
     with gr.Row():
         questions_output = gr.Textbox(label="Generated Questions", lines=5)
@@ -148,11 +153,9 @@ with gr.Blocks() as demo:
 
     generate_btn.click(
         generate_questions,
-        inputs=[context_input, num_questions_input, question_type_input],
+        inputs=[context_input, num_questions_input, question_type_input,dropdown],
         outputs=[questions_output, answers_output],
     )
 
 # Launch the app
-demo.launch()
-
-
+demo.launch()