o
    sgA                     @   s   d Z ddlmZ ddlmZ ddlmZ ddlmZ eddgdd	d
iig dZ	e	
d	ddededefddZe	
ddefddZe	
ddefddZedkr`ddlZee	 dS dS )z*Fast API module for the recommender system    )	APIRouter)AcquisitionsRecommender)PERecommendation)
url_parserz/recommenderrecommenderi  descriptionz	Not found)prefixtags	responsesdependenciesz/recommendations/Ncompany_urlacq_directionpublicc                    s,   t | }t }||j||}||S z!Get recommendations for a company)r   
parsed_urlr   get_recommendationsurl*transform_recommender_output_for_front_end)r   r   r   company_url_parsedrecrecommender_output r   3/var/www/html/XCapMarket/controllers/recommender.pyr      s   

r   z/recommendations/funds/c                    (   t | }t }||j}||S r   )r   r   r   r   r   r   r   r   r   r   r   r   r   get_fund_recommendations    
   

r   z/public_comps/c                    r   )z"Get public companies for a company)r   r   r   get_public_compsr   r   r   r   r   r   r   )   r   r   __main__)NN)__doc__fastapir   -services.recommender.acquisitions_recommenderr   &services.recommender.pe_recommendationr   utilsr   recommendationpoststrboolr   r   r   __name__uvicornrunr   r   r   r   <module>   s8    
	