Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
 from backend import get_question
 
-def generate(context, qtype, difficulty, num, use_beam_search, num_beams):
+# def generate(context, qtype, difficulty, num, use_beam_search, num_beams): -----------------------
+def generate(context, qtype, difficulty, num):
     try:
-        return "\n\n".join(get_question(qtype, difficulty, context, "", num, use_beam_search=use_beam_search, num_beams=num_beams))
+        # return "\n\n".join(get_question(qtype, difficulty, context, "", num, use_beam_search=use_beam_search, num_beams=num_beams)) -------------
+        return "\n\n".join(get_question(qtype, difficulty, context, "", num))
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -18,23 +20,22 @@ with gr.Blocks() as demo:
         difficulty = gr.Dropdown(["easy", "medium", "hard"], label="Difficulty")
         num = gr.Slider(1, 5, step=1, label="Number of Questions", value=1)
         
-    # with gr.Row(visible=True) as beam_controls:
+    # with gr.Row(visible=True) as beam_controls:--------------------------------------
     #     use_beam = gr.Checkbox(label="Use Beam Search", value=False)
     #     beam_dropdown = gr.Dropdown(choices=[3, 4, 5, 6], value=3, label="Number of Beams")
 
     # def toggle_beam_controls(n):
     #     return gr.update(visible=(n == 1))
 
-    # num.change(fn=toggle_beam_controls, inputs=num, outputs=beam_controls)
-    use_beam = False
-    beam_dropdown = 1
+    # num.change(fn=toggle_beam_controls, inputs=num, outputs=beam_controls) -------------------------------
 
     with gr.Row():
         btn = gr.Button("Generate Questions")
 
     output = gr.Textbox(label="Generated Questions", lines=10)
 
-    btn.click(fn=generate, inputs=[context, qtype, difficulty, num,  use_beam, beam_dropdown], outputs=output)
+    # btn.click(fn=generate, inputs=[context, qtype, difficulty, num,  use_beam, beam_dropdown], outputs=output)
+    btn.click(fn=generate, inputs=[context, qtype, difficulty, num], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()