aigenrec/luminabackend
0
1�
2 hn5i � � � d dl mZmZmZ d dlmZmZ d dlmZ d dl m3Z4 d dl m Z d dl
m
Z
d dlmZ d dlmZ d d lmZ G d5� d � Z e� Zy )
� )�Dict�Any�List)�
create_client�Client)�settings)�logger)�uuid4)�datetime)� llm_service)�embedding_service)�qdrant_servicec �z � e Zd Zd� Zdededeeef fd�Zdedededeeef fd�Z d
ded ed6ed e e def7d �Z8y)� NotesServicec �^ � t t j t j � | _ y )N)r r � SUPABASE_URL�SUPABASE_SERVICE_KEY�client)�selfs �+D:\lumina\backend\services\notes_service.py�__init__zNotesService.__init__ s � � +� � !� !� � )� )� 9�� � �10project_id�user_id�returnc � � K � | j j d� j d� j d|� j d|� j � }|j11 r*|j12 d }|d |d |d |d |d |d d13�S y # t $ r'}t j d t |� � �� � d }~ww xY w�w)
zGet notes for a project�notes�*r r r �id�content�14created_at�15updated_at)r r r r r! r" NzError getting notes: )16r �table�select�eq�execute�data� Exceptionr �error�str)r r r �response�note�es r � get_noteszNotesService.get_notes s� � �� � ��{�{�(�(��1�8�8��=�@�@� �j���b��G�$�W�W�Y�
� �}�}� �}�}�Q�'�� �t�*�"&�|�"4� #�I�� #�I��"&�|�"4�"&�|�"4�
� � ��� � �L�L�0��Q���9� :� �� �s) �C�B
B �C� C� "B>�>C�Cr c � �� K � | j ||� � d{ ��� }|r�| j j d� j |t j17 � j
� d�� j d|d � j � }t j d|� �� nht t � � }| j j d� j ||||d�� j � }t j d|� �� |j r|j d S i S 7 ��# t $ r'}t j d t |� � �� � d}~ww xY w�w)18z$Create or update notes for a projectNr )r r" r zUpdated notes for project )r r r r zCreated notes for project r z Error creating/updating notes: )r. r r# �updater �utcnow� isoformatr% r& r �infor* r19 �insertr' r( r) )r r r r �existingr+ �note_idr- s r �create_or_update_notesz#NotesService.create_or_update_notes* s7 � �� � � !�^�^�J�� @�@�H�� �;�;�,�,�W�5�<�<� &�"*�/�/�"3�"=�"=�"?�>� � �2�d�H�T�N�+�G�G�I� �20 � � � 8�� � E�F� �e�g�,�� �;�;�,�,�W�5�<�<�!�",� &� &� >� �21 �7�9� � � � � 8�� � E�F�'/�}�}�8�=�=��#� <�"� <�- A��0 � � �L�L�:�3�q�6�(�C� D� �� �sE �E"�D/ �D,�DD/ �)E"�*D/ �+E"�,D/ �/ E �8"E�E � E"N� note_type�topic�selected_documentsc � �� K � t j d|� d|� d|� �� g }|r22||� d|� �g}nd|v rg d�}n d|v rg d�}n|g}d |� �}g }t � }|D ]z } t j | � � d23{ ��� }24t j ||25d |rd |ind26�
� � d27{ ��� } | D ]/ } | d |vs� |j | � |j | d � �1 �| |sydj |d28d D � cg c] } | d �� c} � }
d|� d|
� d�}d|d�g}t j |dd�� � d29{ ��� }|S 7 ��7 ��c c} w 7 �# t $ r'}t j dt |� � �� � d30}~ww xY w�w)zGenerate notes using AIzGenerating notes (z) for project z , topic: z of �Summary)zoverview of the documentzmain concepts and themeszconclusion and resultsz31Key Points)zimportant definitionsz
key takeawayszcritical points�project_N�32 � document_ids)�collection_name� query_vector�limit�filter_conditions�textz#No content found to generate notes.z33 34� z
Generate a **z9** based on the following content.35 36Content:37z�38 39Requirements:40- Use clear, professional Markdown formatting.41- Use headers, bullet points, and bold text for readability.42- Be comprehensive but concise.43- Structure it as a study guide or note set.44 45Respond ONLY with the Markdown content.�user)�roler g �?i� )� temperature�46max_tokenszError generating notes: )r r3 �setr
�generate_embeddingr �search�append�add�joinr �chat_completionr( r) r* )r r r8 r9 r: �queriesr@ �all_hits�47seen_texts�q� embedding�results�hit�context�prompt�messagesr+ r- s r �generate_noteszNotesService.generate_notesO s� � �� �; � �K�K�,�Y�K�~�j�\�QZ�[`�Za�b� c� �G�� �Y�K�t�E�7�";�<���i�'�l�� ��*�W��$�+�� (�� � 5�O��H� ��J� ��"3�"F�"F�q�"I� I� � .� 5� 5�$3�!*� �N`�~�7I�&J�fj� !� �� #�C��6�{�*�4� ����,�"���s�6�{�3� #� � �<� �k�k�(�3�B�-�"H�-�3�3�v�;�-�"H�I�G� '�y�k� 2� � � 48(� +�F� "(�F�;�<�H� (� 8� 8��s�_c� d�d�H��O�G J���� #I�" e�� � � �L�L�3�C��F�8�<� =� �� �s} �E,�A/D9 �3D.�4&D9 �D0�D9 �,+D9 �E,�D9 �* D2�61D9 �'D7�(D9 �-E,�.D9 �0D9 �2D9 �9 E)�"E$�$E)�)E,)NN) �__name__�49__module__� __qualname__r r* r r r. r7 r r[ � r r r r s� � �50� �#� �� ��S�#�X�� �0#��#� �#� � #�51 52�c�3�h� � #�R �(,� C��C� �C� � C�53 !��I� C� 54
�
Cr r N)�typingr r r �supabaser r �config.settingsr � utils.loggerr �uuidr55 r �services.llm_servicer �services.embedding_servicer
�services.qdrant_servicer r �
notes_servicer_ r r �<module>ri s9 �� "� "� *� $� � � � ,� 8� 2�G� G�R ��
r