import gradio as gr
from rag import get_best_answer

# Custom CSS for the interface
css = """
#chatbot { 
    height: 350px; 
    overflow: auto; 
    border-radius: 10px;
    border: 1px solid #e0e0e0;
}
.textbox {
    border-radius: 20px !important;
    padding: 12px 20px !important;
}
.btn-column {
    display: flex;
    flex-direction: column;
    gap: 10px;
}
"""

def create_interface():
    with gr.Blocks(css=css, theme="soft") as demo:
        gr.Markdown("""
            <h1 style='text-align: center;'>University of Education Lahore Chatbot</h1>
            <p style='text-align: center;'>Official AI Assistant for University Information</p>
        """)

        # Define the chat interface
        chatbot = gr.Chatbot(elem_id="chatbot")
        examples = [
            "What are the admission requirements?",
            "How can I contact the administration?",
            "What programs are offered?"
        ]

        with gr.Row():
            message = gr.Textbox(
                label="Type your question here",
                placeholder="Ask about admissions, programs, or university services...",
                elem_classes="textbox",
                scale=4
            )
            with gr.Column(scale=1, elem_classes="btn-column"):
                submit_button = gr.Button("↩️ Enter")
                reset_button = gr.Button("🗑️ Reset Chat")

        # Set up both Enter key and button to trigger the response
        def respond(message, chat_history):
            bot_message = get_best_answer(message)
            chat_history.append((message, bot_message))
            return "", chat_history

        message.submit(respond, [message, chatbot], [message, chatbot])
        submit_button.click(respond, [message, chatbot], [message, chatbot])

        # Reset button to clear history
        def reset_conversation():
            return []
        
        reset_button.click(reset_conversation, [], [chatbot])

        gr.Examples(examples, inputs=message)

    return demo

if __name__ == "__main__":
    demo = create_interface()
    demo.launch()