9farccontioshi/TranscriptApi
0
1o
2 y܊d� � @ s� d dl mZmZmZ d dlmZmZ d dlmZm Z m3Z4m Z m Z d dl
mZmZ d dlmZ d dlZd dlZede�Zee�ZG dd � d e�Ze�ed5� G d d � d e�Ze�ed
� G dd� de�Ze�ed� G dd� de�Ze�ed� dS )� )� Blueprint�request� current_app)�Api�Resource)�title�summarize_youtube_video�summarize_file�summarize_string�answer)� VideoSummary� FileSummary)�dbN� resourcesc @ � e Zd Zdd� ZdS )�VideoTranscriptc 6 C s� t t� tjj|d��� }|d ur||j|jd�dfS zt|�}W n ddidf Y S z/td| d�}t||d |� d7|d � �|d d
�}t j8� |� t j9� � |||d d�dfW S t
yo } zW Y d }~dS d }~ww )N��video_id)r r �summary�� �errorzVideo ID not valid� � https://www.youtube.com/watch?v=�TranscriptApi/common/audio�The title of the video is �. �10transcriptr �r r r r �� )�printr r �query� filter_by�firstr r r r �session�add�commit� Exception)�selfr Z summaryExist� video_title�output�newVideo�e� r, �Fd:\Coding\Main_Project\TranscriptApi\TranscriptApi\resources\routes.py�get s$ $ 11��zVideoTranscript.getN)�__name__�12__module__� __qualname__r. r, r, r, r- r
� r z /video_api/<string:video_id>c @ r )�FileTranscriptc C sF |dks|dkr9t tj� tjd }tj�tj�d�|j �}|�13tj�tj�d�|j �� t ||d�\}}|j }n|dkrHt tj
d �\}}d}|d krntj�tj�d��r_t�tj�d�� t�tj�d�� d14d id fS t|||d
�}tj�|� tj�� tj�tj�d��r�t�tj�d�� t�tj�d�� ||d�dfS )NZpdf�txt�file�
UPLOAD_FOLDER)�
file_locationZfile_extensionZ direct_text�textz Entered Textz[ERROR]r z#We are expreriencing some issues...r )r r r )r r r )r r �files�os�path�joinr �configr. �filename�saver r15 �json�exists�shutil�rmtree�mkdirr
r r# r$ r% )r' �typer5 r7 r r � file_nameZ16newSummaryr, r, r- �post# s, 1718 19zFileTranscript.postN�r/ r0 r1 rG r, r, r, r- r3 " r2 r3 z/file_api/<string:type>c @ r )�VideoQuestionsc 20 C s� t tj� tjj|d��� }|d u r-td| d�\}}t|�}t||d|� d|� �|d�}tjj|d��� }tj}zt |d |j21d�} |d | d �d22fW S d d id
f Y S )Nr r r r r r �question�rJ �context�rJ r r r zsomething went wrongr ) r r r@ r r r! r" r r r r )23r' r Z videoExistsr r r( r* Z VideoExists�data�ansr, r, r- rG ? s 24 zVideoQuestions.postNrH r, r, r, r- rI > r2 rI z%/video_question_api/<string:video_id>c @ r )�
FileQuestionsc C sV t jj|d��� }t|� |d ur%ttjd |jd�}tjd |d�dfS ddidfS ) N)�idrJ rK rM r r zfile not foundr ) r
r r! r" r r r r@ r )r' rQ ZtranscriptDatarO r, r, r- rG R s zFileQuestions.postNrH r, r, r, r- rP Q r2 rP z/file_question_api/<int:id>) �flaskr r r Z
flask_restfulr r ZTranscriptApi.common.utilsr r r r25 r ZTranscriptApi.modelsr r
�
TranscriptApir r: rB r/ r �apir Z add_resourcer3 rI rP r, r, r, r- �<module> s" 26