Add comment for gen LP

main.py

@@ -429,7 +429,7 @@ LEGAL_POSITION_SCHEMA = {
 }
 
 
-def generate_legal_position(court_decision_text, user_question):
+def generate_legal_position(court_decision_text, comment_input):
     """
     Генерує правову позицію на основі тексту судового рішення.
 
@@ -447,13 +447,17 @@ def generate_legal_position(court_decision_text, user_question):
             temperature=0
         )
 
+        # Формування повідомлень для чату
+        # Формуємо контент з урахуванням коментаря
+        content = LEGAL_POSITION_PROMPT.format(
+            court_decision_text=court_decision_text,
+            comment=comment_input if comment_input else "Коментар відсутній"
+        )
+
         # Формування повідомлень для чату
         messages = [
             ChatMessage(role="system", content=SYSTEM_PROMPT),
-            ChatMessage(
-                role="user",
-                content=LEGAL_POSITION_PROMPT.format(court_decision_text=court_decision_text)
-            ),
+            ChatMessage(role="user", content=content),
         ]
 
         # Отримання відповіді від моделі
@@ -489,179 +493,11 @@ def generate_legal_position(court_decision_text, user_question):
         }
 
 
-# def create_gradio_interface():
-#     with gr.Blocks() as app:
-#         gr.Markdown("# Аналізатор релевантних Правових Позицій Верховного Суду для нового судового рішення")
-#
-#         with gr.Row():
-#             url_input = gr.Textbox(label="URL судового рішення:")
-#             question_input = gr.Textbox(label="Уточнююче питання для аналізу:")
-#
-#         with gr.Row():
-#             generate_position_button = gr.Button("Генерувати короткий зміст позиції суду")
-#             search_with_ai_button = gr.Button("Пошук із ШІ", interactive=False)
-#             # search_without_ai_button = gr.Button("Пошук без ШІ")
-#             analyze_button = gr.Button("Аналіз", interactive=False)
-#
-#         position_output = gr.Markdown(label="Короткий зміст позиції суду за введеним рішенням")
-#         search_output = gr.Markdown(label="Результат пошуку")
-#         analysis_output = gr.Markdown(label="Результат аналізу")
-#
-#         # Два об'єкти стану для зберігання legal_position_json та nodes
-#         state_lp_json = gr.State()
-#         state_nodes = gr.State()
-#
-#         async def generate_position_action(url):
-#             try:
-#                 court_decision_text = extract_court_decision_text(url)
-#                 legal_position_json = generate_legal_position(court_decision_text, "")
-#                 position_output_content = f"**Короткий зміст позиції суду за введеним рішенням:**\n *{legal_position_json['title']}*: \n{legal_position_json['text']} **Категорія:** \n{legal_position_json['category']} ({legal_position_json['proceeding']})\n\n"
-#                 return position_output_content, legal_position_json
-#             except Exception as e:
-#                 return f"Error during position generation: {str(e)}", None
-#
-#         async def search_with_ai_action(legal_position_json):
-#             try:
-#                 query_text = legal_position_json["title"] + ': ' + legal_position_json["text"] + ': ' + legal_position_json["proceeding"] + ': ' + legal_position_json["category"]
-#                 nodes = await retriever_bm25.aretrieve(query_text)
-#
-#                 sources_output = "\n **Результати пошуку (наявні правові позиції ВСУ):** \n\n"
-#                 for index, node in enumerate(nodes, start=1):
-#                     source_title = node.node.metadata.get('title')
-#                     doc_ids = node.node.metadata.get('doc_id')
-#                     lp_ids = node.node.metadata.get('lp_id')
-#                     links = get_links_html(doc_ids)
-#                     links_lp = get_links_html_lp(lp_ids)
-#                     sources_output += f"\n[{index}] *{source_title}* {links_lp} 👉 Score: {node.score} {links}\n"
-#
-#                 return sources_output, nodes
-#             except Exception as e:
-#                 return f"Error during search: {str(e)}", None
-#
-#         async def search_without_ai_action(url):
-#             try:
-#                 court_decision_text = extract_court_decision_text(url)
-#                 nodes = await retriever_bm25.aretrieve(court_decision_text)
-#
-#                 search_output_content = f"**Результати пошуку (наявні правові позиції ВСУ):** \n\n"
-#                 for index, node in enumerate(nodes, start=1):
-#                     source_title = node.node.metadata.get('title', 'Невідомий заголовок')
-#                     doc_ids = node.node.metadata.get('doc_id')
-#                     links = get_links_html(doc_ids)
-#                     search_output_content += f"\n[{index}] *{source_title}* 👉 Score: {node.score} {links}\n"
-#
-#                 return search_output_content, nodes
-#             except Exception as e:
-#                 return f"Error during search: {str(e)}", None
-#
-#         async def analyze_action(legal_position_json, question, nodes):
-#             try:
-#                 # Використання з OpenAI
-#                 workflow = PrecedentAnalysisWorkflow(
-#                     provider=ModelProvider.OPENAI,
-#                     model_name=ModelName.GPT4_MINI
-#                 )
-#
-#                 # # Використання з Anthropic
-#                 # workflow_anthropic = PrecedentAnalysisWorkflow(
-#                 #     provider=ModelProvider.ANTHROPIC,
-#                 #     model_name=ModelName.CLAUDE3_SONNET
-#                 # )
-#
-#                 # Формуємо єдиний текст запиту з legal_position_json
-#                 query = (
-#                     f"{legal_position_json['title']}: "
-#                     f"{legal_position_json['text']}: "
-#                     f"{legal_position_json['proceeding']}: "
-#                     f"{legal_position_json['category']}"
-#                 )
-#
-#                 # Запускаємо workflow і отримуємо текст аналізу
-#                 response_text = await workflow.run(
-#                     query=query,
-#                     question=question,
-#                     nodes=nodes
-#                 )
-#
-#                 # Формуємо вивід
-#                 output = f"**Аналіз ШІ:**\n{response_text}\n\n"
-#                 output += "**Наявні в базі Правові Позицій Верховного Суду:**\n\n"
-#
-#                 # Розбиваємо текст відповіді на рядки
-#                 analysis_lines = response_text.split('\n')
-#
-#                 # Проходимо по кожному рядку аналізу
-#                 for line in analysis_lines:
-#                     if line.startswith('* ['):
-#                         # З кожного рядка отримуємо індекс
-#                         index = line[3:line.index(']')]  # Витягуємо індекс з "* [X]"
-#
-#                         # Знаходимо відповідний node за індексом
-#                         node = nodes[int(index) - 1]
-#                         source_node = node.node
-#
-#                         source_title = source_node.metadata.get('title', 'Невідомий заголовок')
-#                         source_text_lp = node.text
-#                         doc_ids = source_node.metadata.get('doc_id')
-#                         lp_id = source_node.metadata.get('lp_id')
-#
-#                         links = get_links_html(doc_ids)
-#                         links_lp = get_links_html_lp(lp_id)
-#
-#                         output += f"[{index}]: *{source_title}* | {source_text_lp} | {links_lp} | {links}\n\n"
-#
-#                 return output
-#
-#             except Exception as e:
-#                 return f"Error during analysis: {str(e)}"
-#
-#         # Підключаємо функції до кнопок з оновленими входами та виходами
-#         generate_position_button.click(
-#             fn=generate_position_action,
-#             inputs=url_input,
-#             outputs=[position_output, state_lp_json]
-#         )
-#         generate_position_button.click(
-#             fn=lambda: gr.update(interactive=True),
-#             inputs=None,
-#             outputs=search_with_ai_button
-#         )
-#
-#         search_with_ai_button.click(
-#             fn=search_with_ai_action,
-#             inputs=state_lp_json,
-#             outputs=[search_output, state_nodes]
-#         )
-#         search_with_ai_button.click(
-#             fn=lambda: gr.update(interactive=True),
-#             inputs=None,
-#             outputs=analyze_button
-#         )
-#
-#         # search_without_ai_button.click(
-#         #     fn=search_without_ai_action,
-#         #     inputs=url_input,
-#         #     outputs=[search_output, state_nodes]
-#         # )
-#         # search_without_ai_button.click(
-#         #     fn=lambda: gr.update(interactive=True),
-#         #     inputs=None,
-#         #     outputs=analyze_button
-#         # )
-#
-#         analyze_button.click(
-#             fn=analyze_action,
-#             inputs=[state_lp_json, question_input, state_nodes],
-#             outputs=analysis_output
-#         )
-#
-#     return app
-
 def create_gradio_interface():
     async def generate_position_action(url):
         try:
             court_decision_text = extract_court_decision_text(url)
-            legal_position_json = generate_legal_position(court_decision_text, "")
+            legal_position_json = generate_legal_position(court_decision_text, comment_input)
             position_output_content = f"**Короткий зміст позиції суду за введеним рішенням:**\n *{legal_position_json['title']}*: \n{legal_position_json['text']} **Категорія:** \n{legal_position_json['category']} ({legal_position_json['proceeding']})\n\n"
             return position_output_content, legal_position_json
         except Exception as e:
@@ -741,6 +577,7 @@ def create_gradio_interface():
         gr.Markdown("# Аналізатор релевантних Правових Позицій Верховного Суду для нового судового рішення")
 
         with gr.Row():
+            comment_input = gr.Textbox(label="Коментар до формування короткого змісту судового рішення:")
             url_input = gr.Textbox(label="URL судового рішення:")
             question_input = gr.Textbox(label="Уточнююче питання для аналізу:")
 

prompts.py

@@ -12,25 +12,32 @@ LEGAL_POSITION_PROMPT = """Дотримуйся цих інструкцій.
 {court_decision_text}
 </court_decision>
 
-2. Уважно прочитай та проаналізуй текст наданого судового рішення. Зверни увагу на:
+2. Врахуй наступний коментар щодо формування правової позиції:
+<comment>
+{comment}
+</comment>
+
+3. Уважно прочитай та проаналізуй текст наданого судового рішення. Зверни увагу на:
    - Юридичну суть рішення
    - Основне правове обґрунтування
    - Головні юридичні міркування
+   - Аспекти, зазначені в коментарі
 
-3. На основі аналізу сформулюй короткий зміст позиції суду, дотримуючись таких вказівок:
+4. На основі аналізу сформулюй короткий зміст позиції суду, дотримуючись таких вказівок:
    - Будь чіткими, точними та обґрунтованими
    - Використовуй відповідну юридичну термінологію
    - Зберігай стислість, але повністю передай суть судового рішення
    - Уникай додаткових пояснень чи коментарів
    - Спробуй узагальнювати та уникати специфічної інформації (наприклад, імен або назв) під час подачі результатів
    - Використовуйте лише українську мову
+   - Врахуй аспекти та зауваження з коментаря при формуванні позиції
 
-4. Створи короткий заголовок, який відображає основну суть судового рішення та зазнач його категорію.
+5. Створи короткий заголовок, який відображає основну суть судового рішення та зазнач його категорію.
 
-5. Додатково визнач тип судочинства, до якої відноситься дане рішення.
+6. Додатково визнач тип судочинства, до якої відноситься дане рішення.
     Використовуй лише один із цих типів: 'Адміністративне судочинство', 'Кримінальне судочинство', 'Цивільне судочинство', 'Господарське судочинство'
 
-6. Відформатуй відповідь у форматі JSON:
+7. Відформатуй відповідь у форматі JSON:
 
 {{
     "title": "Заголовок судового рішення",