Update app.py

app.py

@@ -1,29 +1,58 @@
-from flask import Flask, render_template, request
+import gradio as gr
+import random
 from llm import get_prompt, get_responce
 
-app = Flask(__name__)
-
-# Assume your question generation logic is already defined
 def generate_questions(question_type):
-    if question_type == 'mcq':
-        return ["What is the capital of France?", "Choose the correct answer from four options."]
-    elif question_type == 'fill_blank':
-        return ["The sun rises in the ____.", "Fill in the correct word."]
-    elif question_type == 'short_answer':
-        return ["Explain the process of photosynthesis."]
-    else:
-        return ["No questions available for the selected type."]
-
-@app.route('/')
-def home():
-    return render_template('index.html')
-
-@app.route('/get_questions', methods=['POST'])
-def get_questions():
-    question_type = request.form['question_type']
     prompt, seed = get_prompt(question_type)
     questions = get_responce(prompt)
-    return render_template('questions.html', questions=questions, question_type=questions[0]["question_type"], seed = seed)
+    return questions, seed
+
+def process_question_type(question_type):
+    questions, seed = generate_questions(question_type)
+    formatted_questions = []
+    
+    for question in questions:
+        question_text = question["question"]
+        options = question.get("options", {})
+        answer = question["answer"]
+        correction = question.get("correction", "")
+        
+        question_display = {
+            "Question": question_text,
+            "Answer": answer,
+            "Correction": correction,
+            "Options": options if options else None,
+        }
+        formatted_questions.append(question_display)
+        
+    return formatted_questions, seed
+
+def main():
+    question_types = [
+        "Choose the correct meaning of the word",
+        "Personal response",
+        "Give reasons",
+        "Fill in the blanks",
+        "True or False",
+    ]
+
+    # Create a Gradio interface
+    with gr.Blocks() as demo:
+        gr.Markdown("<h1>Select Question Type</h1>")
+        question_type = gr.Dropdown(choices=question_types, label="Question Type", value=question_types[0])
+        generate_button = gr.Button("Generate Questions")
+        
+        output = gr.Textbox(label="Generated Questions", interactive=False)
+        seed_output = gr.Textbox(label="Seed Number", interactive=False)
+        
+        def generate():
+            questions, seed = process_question_type(question_type.value)
+            formatted_output = "\n".join([f"{q['Question']} - Answer: {q['Answer']}, Correction: {q['Correction']}" for q in questions])
+            return formatted_output, seed
+        
+        generate_button.click(generate, outputs=[output, seed_output])
+        
+    demo.launch()
 
-if __name__ == '__main__':
-    app.run(debug=True)
+if __name__ == "__main__":
+    main()