AlvesS077/SmartPlate
0
1�
2 r"h� � �� � S SK JrJrJrJrJr S SKJr S SKrS SK r S SK3r4S SK J r J
r
S SKJr S SKJr S SKr\" \SSS 9r\ R* " S5S 9r\6R. R0 " \" \R5 5 5 SS 6 r\R7 5 S
r SS jr \R= S5 S 5 r \R= SS/S9S 5 r \R= S5 S 5 r!\S:X a \RE S7S9 gg)� )�Flask�request�jsonify�render_template� send_file)� HTMLExporterN)�models�8transforms)�Image)� normalize�.� )�
static_folder�static_url_pathT)�9pretrained�����c �( � [ R " [ R " S5 [ R " 5 [ R " / SQ/ SQS9/5 n[10 R " U 5 R S5 nU" U5 R S5 n[ R " 5 [ U5 nS S S 5 [ WSSS 9nUR 5 U4$ ! , ( d f N+= f! [ a n[ S11U S U 35 S nAg S nAff = f)
N)�� r )g12ףp=13�?g�v��/�?g�C�l���?)g Zd;�O�?gy�&1��?g�������?)�mean�std�RGBr � � )�p�dimzErro ao processar a imagem z: )NN)r14 �Compose�Resize�ToTensor� Normalizer �open�convert� unsqueeze�torch�no_grad�modelr �squeeze� Exception�print)� imagem_path�15preprocess�img�16img_tensor� embedding�es �.C:\Users\Admin\Desktop\projeto economia\app.py�
get_embeddingr0 s� � �� �'�'� � � �j� )� � � � !� � � �&;�AV� W�)17� �18� �j�j��%�-�-�e�4�� ��_�.�.�q�1�19�
�]�]�_� �j�)�I� � �i�1�!�4� �� � �"�C�'�'� �_��20 � �
�+�K�=��1�#�>�?����s0 �B(C, �* C�6$C, �21C)�%C, �,22D�6D � Dc �� � [ U 5 u p4Uc g/ n[ R " U5 H� u pgnU H� n U R 5 R S5 ( d M) [ R23 R
Xi5 n24[ U255 u p�U c MZ S[ R " X;SS9R 5 -26 n [ R27 R [ R28 R U295 5 n
UR X�U
45 M� M� UR S S9 US U n/ nU H8 u np�UR [ R30 R U5 U
U S.5 M: US 4$ ) N)Nu9 Não foi possível extrair embedding da imagem fornecida.)�jpg�jpeg�png�bmpr r )r c � � U S $ )Nr � )�xs r/ �<lambda>�#comparar_com_base.<locals>.<lambda>7 s � �!�A�$� )�key)�31image_name�32class_name�distance)r0 �os�walk�lower�endswith�path�joinr# �cosine_similarity�item�basename�dirname�append�sort)�imagem_usuario_path�base_dir�top_k� emb_usuario� img_usuario�33resultados�root�_�files�fname�img_path�emb_base�distr> �top_resultados�responserD s r/ �comparar_com_baser[ % s< � � ,�-@� A� �K���P��J� �'�'�(�+�����E��;�;�=�)�)�*G�H�H���w�w�|�|�D�0�H�'��1�K�H�� ���u�.�.�{�!�L�Q�Q�S�S�D����)�)�"�'�'�/�/�(�*C�D�J� � � �x�z� :� ;� � ,� �O�O��O�'� ���'�N� �H�"0� ��d���� �'�'�*�*�4�0�$� �34� � #1� �T�>�r; �/c �, � [ R S5 $ )Nz35index.html)�app�send_static_filer7 r; r/ � serve_indexr` E s � � � � � � -�-r; z
/analyze-food�POST)�methodsc � � S[ R ; a [ SS05 S4$ [ R S n U R S:X a [ SS05 S4$ [ R36 R
SU R 5 n[ R " SSS 9 U R U5 [ R37 R S385 n[ X5 u p4[ R " U5 U( a [ SU05 S4$ [ S U05 $ ) N�image�errorzNo image file providedi� r zNo selected file�tempT)�exist_ok�train�results) r rT r �filenamer@ rD rE �makedirs�save�abspathr[ �remove)�39image_file� temp_pathrM ri re s r/ � analyze_foodrq I s� � ��g�m�m�#���!9�:�;�S�@�@�� � �w�'�J����b� ���!3�4�5�s�:�:� ��� � �V�Z�%8�%8�9�I��K�K���&��O�O�I� ��w�w���w�'�H�&�y�;�N�G��I�I�i�� ����'�(�#�-�-� �I�w�'� (�(r; z
/run-notebookc �� � Sn [ U SSS9 n[ R " USS9nS S S 5 [ 5 nUR W5 u pEU$ ! , ( d f N-= f! [40 a nSU S3s S nA$ S nAff = f) Nz final.ipynb�rzutf-8)�encoding� )�41as_versionz%<h1>Erro ao abrir notebook:</h1><pre>z</pre>)r �nbformat�readr �from_notebook_noder' )�caminho_notebook�f�notebook�
html_exporter�bodyrS r. s r/ � ver_notebookr ` s � � %�� A�
�"�C�'�
:�a� �}�}�Q�1�5�H� ;� %��
� �2�2�8�<���� �
;�
:�� � A�6�q�c��@�@��A�s3 � A �A �&A � 42A�A � 43A7�'A2�,A7�2A7�__main__)�debug)� )#�flaskr r r r r � nbconvertr rw r@ r# � torchvisionr r44 �PILr �torch.nn.functionalr �io�__name__r^ �resnet50r% �nn�45Sequential�list�children�evalr0 r[ �router` rq r �runr7 r; r/ �<module>r� s� �� E� E� "� � � � *� � )� � �H�C��<�� ���4�(��
�����d�5�>�>�#3� 4�S�b� 9�;�� �46�47� ��& �@ ���3� �.� �.� ���?�V�H��-�)� .�)�, ���?��
A� �
A� �z���G�G�$�G�� r;