import gradio as gr
from ai_assistant import get_response

def respond(message, history):
    print("📝 User query:", message)
    response = get_response(message)
    print("🤖 Response:", response)
    return response

chatbot = gr.ChatInterface(
    fn=respond,
    type="messages",
    title="Research Query Assistant",
    description="Ask research questions and get answers from Mistral-7B with 8-bit quantization.",
)

if __name__ == "__main__":
    chatbot.launch(share=True)