main.py

import os
import pathlib
import textwrap
import google.generativeai as genai
# from google.colab import userdata # Used to securely store the API key
# from IPython.display import display
from IPython.display import Markdown
import warnings
warnings.filterwarnings("ignore")


os.environ["GOOGLE_API_KEY"] = "AIzaSyDeEFxbAE5isnFWJK-cliU7tAS_mOjTan4"
genai.configure(api_key=os.environ["GOOGLE_API_KEY"])

model = genai.GenerativeModel("gemini-pro")


def to_markdown(text):
    text = text.replace(".", "* ")
    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))

# Define MCQ generator function
def mcq_generator(programming, level, number_of_questions):
    response = model.generate_content(f"Write {number_of_questions} MCQ questions for language {programming} with level {level} and answer also.")
    return response.text