# Step 2: Import necessary libraries
import gradio as gr
from google.colab import files
from llama_index.readers.file import PDFReader
from llama_index.core import VectorStoreIndex

# Step 3: Upload PDF
uploaded = files.upload()
filename = next(iter(uploaded))

# Step 4: Read and index the PDF
reader = PDFReader()
documents = reader.load_data(file=filename)
index = VectorStoreIndex.from_documents(documents)
query_engine = index.as_query_engine()

# Step 5: Define the chatbot function
def ask_resume_bot(question):
    response = query_engine.query(question)
    return str(response)

# Step 6: Launch Gradio UI
iface = gr.Interface(
    fn=ask_resume_bot,
    inputs=gr.Textbox(placeholder="Ask something about your resume...", label="Your Question"),
    outputs=gr.Textbox(label="Bot's Response"),
    title="Resume Chatbot",
    description="Ask questions about your resume like skills, experience, education, etc."
)
iface.launch()