filename:
__pycache__/server.cpython-310.pyc
branch:
main
back to repo
o
hQ�i& � @ s� d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZe� Z e j
dedd�dd� e j
d ed
d�d
d� e � d �d d
� �Z dd� Z
dd� Ze �d�defdd��ZdS )� )�FastAPI�� FileResponse)� StaticFilesNz/staticZstatic)� directory)�namez/promptsZprompts�/c C s t d�S )Nzstatic/index.htmlr � r r �;C:\Users\Ralpo\OneDrive\Desktop\research-frontend\server.py�root s r c C s8 t dd��
} t�| �W d � S 1 sw Y d S )Nz config.json�r)�open�json�load)�fr r r
� load_config s $�r c
� s� �|d }|d }|d }|� d|� �}zQ| j d|� d�||dd�d d
�4 I d H �,}|�� d }|�� 2 z3 d H W } | rIt�| �}
||
�d d �7 }q46 W d �I d H n 1 I d H s[w Y ||d
�W S ty~ } z
|t| �d�W Y d } ~ S d } ~ ww )N�urlr Znotez
MODEL NOTE: �POSTz
/api/generateT)�model�prompt�stream�x )r �timeout� �response)r �output)r �error)r Zraise_for_statusZ aiter_linesr �loads�get� Exception�str) �clientZ model_config� full_promptZ model_urlZ
model_nameZ
model_noteZaugmented_promptr r �line�data�er r r
� run_model sD ���
��(�����r& z /api/run-all�payloadc � s� �| � dd�}tdd�� }|�� }W d � n1 sw Y d|� d|� ��t� }|� dg �}t�� 4 I d H � � � �fdd �|D �}tj|� I d H }W d �I d H |S 1 I d H s]w Y |S )
Nr r z prompts/motion-description.txtr zSYSTEM:
z
USER:
�modelsc s g | ]}t � |���qS r )r& )�.0r �r! r" r r
�
<listcomp>L s
��zrun_all.<locals>.<listcomp>)r r
�readr �httpxZ AsyncClient�asyncio�gather)r' r r �context�configr( �tasks�resultsr r* r
�run_all> s" �
� ���r4 )Zfastapir Zfastapi.responsesr Zfastapi.staticfilesr r- r. r �appZmountr r r r& Zpost�dictr4 r r r r
�<module> s
&