Resume-qna-bot / app.py
jefalod's picture
Update app.py
b37ef29 verified
raw
history blame contribute delete
943 Bytes
# 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()