from google import genai
import os
from google.genai import types
import gradio as gr

client = genai.Client(api_key = os.getenv('GEMINI_API_KEY'))

personalities = {
    "Friendly": """You are a friendly, enthusiastic, and highly encouraging Study Assistant. Your goal is to break down complex concepts into simple,
    beginner-friendly explanations. Use analogies and real-world examples that beginners can relate to. Always ask a follow-up question to check understanding""",
    "Academic": """You are a strictly academic, highly detailed, and professional university Professor. Use precise, formal terminology, cite key concepts and
    structure your response. Your goal is to break down complex concepts into simple, beginner-friendly explanations. Use analogies and real-world examples that
    beginners can relate to. Always ask a follow-up question to check understanding"""
}

def study_assistant(question, persona):
  system_prompt = personalities[persona]
  response = client.models.generate_content(
    model = "gemini-2.5-flash", contents = question,
    config = types.GenerateContentConfig(system_instruction = system_prompt)
    )
  return response.text

demo = gr.Interface(fn = study_assistant,
             inputs = [gr.Textbox(label = "Question", lines = 4, placeholder = "Ask a Question"),
                       gr.Radio(choices= list(personalities.keys()), value= "Friendly", label= "Personality")],
             outputs = gr.Textbox(label = "Answer or Explaination", lines=10),
             title = "Study Assistant",
             description= "Ask a quesiton to get simple explaination")

demo.launch(debug= True)