Hamid Omarov
Space: add langchain-community for PyPDFLoader
b3617c5
raw
history blame
738 Bytes
import sys
import os
import gradio as gr
# day3 qovluğunu sys.path-ə əlavə et
sys.path.append(os.path.join(os.path.dirname(__file__), "day3"))
from rag_system import RAGPipeline
rag = RAGPipeline()
def ask_pdf(pdf_file, question):
if pdf_file:
info = rag.index_document(pdf_file.name)
else:
info = {"chunks_indexed": 0, "best_strategy": None}
out = rag.query(question)
return f"**Answer:** {out['answer']}\n\n_Used chunks: {out['used_chunks']}_"
demo = gr.Interface(
fn=ask_pdf,
inputs=[
gr.File(label="Upload PDF", file_types=[".pdf"]),
gr.Textbox(label="Question")
],
outputs="markdown",
title="PDF RAG System"
)
if __name__ == "__main__":
demo.launch()