FacundoSander/PdfQA
1
1o
2 އUde � @ s� d dl mZmZmZmZmZ d dlmZ d dlm Z d dl3m Z m Z m
Z
d dlmZ d dl mZ d dlZe� Zejded d4�d d � e�d �d
d� �Ze�d�ed�ed�ed�ed�ed�fdededededef5dd��ZdS )� )�FastAPI�File�6UploadFile�Form�
HTTPException)� FileResponse)�escape)�create_qa_object�7get_answer�get_source_documents)� StaticFiles)r Nz/staticZstatic)� directory)�name�/c � s( �zt d�W S ty tddd��w )Nzstatic/index.htmli� zFile not found)� status_codeZdetail)r �FileNotFoundErrorr � r r �GC:\Users\facun\OneDrive\Escritorio\aprendiendo\aprendiendo\webQA\api.py�get_index_html
s �8 �r z/ask.�stuff� �api_key�file�query�9chain_type�kc 10 � s| �| t jd< tdd��}|�|�� I d H � W d � n1 s w Y td||�}t||�}t||�}t|�} || d�S )NZOPENAI_API_KEYztemp.pdf�wb)�answer�source) �os�environ�open�write�readr r11 r r )12r r r r r �bufferZqar r Zescaped_sourcer r r �ask s �13 � 141516r% )Zfastapir r r r r Zfastapi.responsesr �htmlr �mainr r17 r Zfastapi.staticfilesr r �appZmount�getr Zpost�str�intr% r r r r �<module> s 18D