Umesh-960's picture
Update app.py
a62ed30 verified
Raw
History Blame Contribute Delete
1.25 kB
import os
from google import genai
from google.genai import types
import gradio as gr
client=genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
Question_type={
"MCQs":"Generate question of MCQs type based upon the content given.",
"Short Answer":"Generate question of short-answer type based upon the content given.",
"Interview":"Generate question of Interview type based upon the content given."
}
def question_generator(content,question_type):
system_prompt=Question_type[question_type]
response=client.models.generate_content(
model="gemini-2.0-flash",
config=types.GenerateContentConfig(
system_instruction=system_prompt,
temperature=0.7,
max_output_tokens=1000
),
contents=content
)
return response.text
demo=gr.Interface(
fn=question_generator,
inputs=[gr.Textbox(lines=4,placeholder="Paste Content/study material...",label="inputContent"),
gr.Dropdown(choices=list(Question_type.keys()),label="question_type",value="MCQs")],
outputs=gr.Textbox(lines=12,label="Generated Questions"),
title="Question Generator",
description="Paste the Content/Study material and get the questions of the selected type."
)
demo.launch(debug=True)