FacundoSander/PdfQA

1o

2�Ud��@s�dd
lm
Z

ddlmZ
ddlmZ
ddlmZ
ddlm	Z	
ddl3mZ
ddlm
Z

ddlmZ
e
�Zejd	ed4d�
d5d�
d
d�Zdd�Zdd�Zedkr]ddlZejddddd�
dSdS)�)
�RetrievalQA)
�OpenAI)
�PyPDFLoader)
�CharacterTextSplitter)
�OpenAIEmbeddings)
�Chroma)
�FastAPI)
�StaticFilesz/static�static)
�	directory)
�namecCsbt|�
}|�
�}td
dd�}|�|�
}t�}t�||�}|jdd|i
d�}	tj	t6�|
|	dd�}7|8S)	Ni�r
)�9chunk_sizeZ
chunk_overlapZ10similarity�
k)Zsearch_typeZ
search_kwargsT)Zllm�11chain_type�	retrieverZreturn_source_documents)r�loadrZsplit_documentsrrZfrom_documentsZas_retrieverrZfrom_chain_typer)�filerr�loaderZ	documentsZ
text_splitterZtextsZ12embeddings�dbr�qa�r�HC:\Users\facun\OneDrive\Escritorio\aprendiendo\aprendiendo\webQA\main.py�create_qa_objects
13
14
�rcCs|d
|
i
�
}|dS)N�query�resultr)rrrrrr�15get_answer&s

rcCsJ|d
|
i
�
}|d}g}|D]}|�|j
dd��d��

qd�|�
}|S)Nr�source_documentsi�z...z | )�appendZpage_content�join)rrrrZsource_info�docZsource_textrrr�get_source_documents*s




16

r �__main__Nzapi:appz0.0.0.0i@T)�host�port�reload)Zlangchain.chainsrZlangchain.llmsrZlangchain.document_loadersrZlangchain.text_splitterrZlangchain.embeddingsrZlangchain.vectorstoresr�fastapir�fastapi.staticfilesr	�app�mountrrr �__name__�uvicorn�runrrrr�<module>
s"






	

�