FastMCP Supply Chain Optimizer

o �\h��@s2dd
lZ
dd
lZddlmZmZ
Gdd�d�Zd
S)�N)�Dict�Anyc@s�eZ
dZddefdd�Zddedeeeffdd �Zded ededeeeffdd �Z dededededeeeff dd�Z ded edeeeffdd�Zdededeeeffdd�Zdd�Z dd�ZdS)�SupplyChainTools�data/inventory.csv�inventory_filecCs|
|_t
�|
�
|_g|_dS�
N)r�pdZread_csv�inventory_df�actions_log)�selfr�r�?/Users/anshriyal/Downloads/github/FastMCP/supply_chain_tools.py�__init__s


zSupplyChainTools.__init__N� product_id�returncCsH|
r|j|jd
|
k}|j
rdd|
�d�i
S|�d�
dS|j�d�
S)z:Get current inventory status for a product or all productsr�error�Product � not found�recordsr
)r �empty�to_dict)rr�product_datarrr �get_inventory_statuss



z%SupplyChainTools.get_inventory_status� warehouse�quantitycCs�|d
v
rdd|��i
S|jd|
k}|�
�sdd|
�d�i
S|jj||fjd}td||�}||jj||f<d|
�d |�d |�d|��}|j�|�

d|
||||d �S)z5Update inventory for a specific product and warehouse)�warehouse_A�warehouse_BZwarehouse_CrzInvalid warehouse: rrrr
zUpdated z in z: z -> T)�successrrZold_quantityZnew_quantity�action)r �any�loc�iloc�maxr �append)rrrr�maskZcurrent_qtyZnew_qtyrrrr �update_inventorys"










�z!SupplyChainTools.update_inventory�from_warehouse�to_warehousecCs~|�|
||�}d
|vr|S|�|
||�}d
|vr"|�|
||�
|Sd|�d|
�d|�d|��}|j
�|�

d|||||
d�S)z;Calculate and execute inventory transfer between warehousesrz TRANSFER: � units of � from z to T)rrr&r'rr)r%r r#)rrr&r'rZresult1�result2rrrr �calculate_transfer.s 









�z#SupplyChainTools.calculate_transfercCs�|j|jd
|
k}|j
rdd|
�d�i
S||jd}|dk
r#d}n|dk
r*d }n |d k
r1d}nd}|
||||dk
r?d d�S|dk
rGdd�Sdd�S)z:Simple prediction of when a product might run out of stockrrrrr
�z1-2 days�2z3-5 days�dz 1-2 weekszMore than 2 weeks�HIGH�MEDIUM�LOW)rr� current_stockZpredicted_stockoutZ risk_level)r rr!)rrrrr2Zdays_to_stockoutrrr �predict_stockoutHs*










� ��z!SupplyChainTools.predict_stockoutcCs�|j|jd
|
k}|j
rdd|
�d�i
S|djd}|djd}||}d|�d |
�d |�d|�d|d �d�}|j�|�

d||
||||d�S)z&Recommend reorder action for a productrrrr�supplierr
� cost_per_unitzORDER: r(r)z at $z/unit (Total: $z.2f�
)T)rrrrr4r5� total_cost)r rr!r r#)rrrrr4r5r7rrrr �recommend_reorderbs 



&






�z"SupplyChainTools.recommend_reorderc

Cs|jj
|jd
d�
dS)z)Save current inventory state back to fileF)
�indexN)r Zto_csvr�
rrrr �save_inventoryyszSupplyChainTools.save_inventoryc


Cs|jS)
zGet all recorded actions)
r r:rrr �get_actions_log}sz SupplyChainTools.get_actions_log)
rr)�__name__� __module__�__qualname__�strrrrr�intr%r+r3r8r;r<rrrr rs
" &r)�pandasr�json�typingrrrrrrr �<module>
s