import gradio as gr
import google.generativeai as genai

# Configure API Key
GOOGLE_API_KEY = "675826031094"
genai.configure(api_key=GOOGLE_API_KEY)

# Initialize the Generative Model
model = genai.GenerativeModel(
    'gemini-1.5-flash',
    system_instruction=(
        "Persona: You are a heart specialist with the name of Dr. Assad Siddiqui. Only provide information related to heart health, symptoms, and advice."
        "Ask users about their heart-related symptoms and provide consultation and guidance based on their input."
        "Always provide brief answers. Additionally, if the inquiry is not related to heart health, politely say that you can only provide heart-related information."
        "Responses should be in Urdu (written in Roman English) and English."
    )
)

# Function to get response from the chatbot
def get_chatbot_response(user_input, chat_history):
    response = model.generate_content(user_input)
    bot_response = response.text.strip()
    
    chat_history.append((user_input, bot_response))
    return chat_history, chat_history

# Gradio Interface
def chat(user_input, history):
    return get_chatbot_response(user_input, history)

with gr.Blocks() as demo:
    gr.Markdown("<h1 style='text-align:center;'>Heart Health Chatbot 🫀</h1>")
    gr.Markdown("<p style='text-align:center;'>Ask Dr. Assad Siddiqui about heart health in both Urdu and English!</p>")
    
    chat_history = gr.State([])
    
    chatbot = gr.Chatbot(label="Heart Health Chatbot").style(height=500)
    user_input = gr.Textbox(placeholder="Type your message here...")
    
    user_input.submit(chat, [user_input, chat_history], [chatbot, chat_history])
    gr.Examples(examples=["What are common heart disease symptoms?", "Mujhe dil ki takleef hai. Kya karoon?"],
                inputs=user_input)
    
    gr.Markdown("<p style='text-align:center;'>Made with ❤️ by Assad Siddiqui</p>")

# Launch the Gradio interface
demo.launch()