import gradio as gr
from rag_pipeline import load_rag_chain

# Load the RAG system
rag_chain = load_rag_chain()

# Interface logic
def ask_question(query):
    result = rag_chain.invoke(query)
    return result['result']

iface = gr.Interface(
    fn=ask_question,
    inputs=gr.Textbox(lines=3, placeholder="Ask something from your PDF...", label="Your Question"),
    outputs="text",
    title="🧠 Custom PDF Chatbot (Flan-T5)",
    description="Ask questions based on your uploaded PDF documents. This app runs entirely using free models."
)

if __name__ == "__main__":
    iface.launch()