Added projectinfo methods

app.py

@@ -319,5 +319,146 @@ def getprojectinfo_route():
     final = final.strip()
     return jsonify({'result': final})
 
+
+@app.route('/getprojectlist', methods=['POST'])
+def getprojectlist_route():
+    data = request.get_json()
+    text = data.get('text', '')
+
+    main_prompts = []
+    prompt1 = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке.
+Найди в тексте ИМЕНА СОБСТВЕННЫЕ участников обсуждения. Кто говорил во время совещания?
+Формат ответа: "Участники: *здесь ФИО говорящих из текста*". Если имя говорящего не указано, то отвечай так: "не указано".
+Текст совещания: "{text}"[/INST]''')
+    
+    prompt2 = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке.
+Представь, что ты лучший в мире смысловик и копирайтер.
+Сделай глубокий вдох и, думая шаг за шагом, выполни задание:
+Каков ход совещания в этом тексте?
+Формат ответа: "Повестка встречи: *здесь текст*".
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Скрипт: {text}[/INST]''')
+    main_prompts.append(prompt1)
+    main_prompts.append(prompt2)
+    
+    main_info =''
+    for i in main_prompts:
+        result = generate(i[:52000])
+        if result is None:
+            return jsonify({'error': 'Сервер LLM временно недоступен. Попробуйте повторить запрос через несколько минут.'})
+        main_info += '\n\n'+result+'\n\n'
+
+    proj_prompt = []
+    prompt = (f'''<s>[INST]Вы отвечаете только на РУССКОМ языке.
+1. Найти все упоминания слов "проект" или "проекты" в тексте.
+2. Прочитать контекст вокруг упоминаний слова "проект" или "проекты" и определить, какие проекты упоминаются.
+3. Составить список названий проектов, которые были найдены в тексте.
+Используй только такой формат ответа: "Проект №: *суть кратко*". Конец ответа.
+Ты больше ничего не говоришь, не комментируешь, не добавляешь.
+Текст: {text}[/INST]''')
+    proj_prompt.append(prompt)
+    
+    list_of_projects =''
+    for i in proj_prompt:
+        result = generate(i[:52000])
+        if result is None:
+            return jsonify({'error': 'Сервер LLM временно недоступен. Попробуйте повторить запрос через несколько минут.'})
+        list_of_projects += result
+    
+    delimiter = 'Проект '
+    proj  = [delimiter+x for x in list_of_projects.split(delimiter) if x]
+    proj = proj[1:]
+    
+    proj_ = []
+    for i in proj:
+        a = i.replace("Проект №", "")
+        a = a.replace("Конец ответа", "")
+        a = a.replace("данный проект", "")  ###убираю слово "проект", чтобы модель не опиралась на него,
+        a = a.replace("проект ", "")             # при ответе на вопрос, проект это или нет
+        a = a.replace('\n', ' ')
+        a = a.replace('</s>', ' ')
+        a = a.strip()
+        proj_.append(a)
+        
+    check_prompts = []
+
+    checking = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке. Сделай глубокий вдох и, думая шаг за шагом, выполни задание:
+У меня есть список проектов. Такие проекты имеются в тексте и являются основными? Список проектов: {proj_}.
+Ты обязательно используешь ТОЛЬКО такой формат ответа:
+"Да: Проект № *название проекта*." ИЛИ "Нет, такого проекта нет".
+Ты больше ничего не говоришь, не комментируешь, не добавляешь.
+Текст: {text}[/INST]''')
+    check_prompts.append(checking)
+
+    real_projects = ''
+    for i in check_prompts:
+        result = generate(i[:52000])
+        if result is None:
+            return jsonify({'error': 'Сервер LLM временно недоступен. Попробуйте повторить запрос через несколько минут.'})
+        real_projects += result
+    
+    real_projects_list = re.findall(r'Да:\s*(.*?)\s*(?:\n\n|$)', real_projects)
+    return jsonify({'result': real_projects_list})
+
+@app.route('/getprojectdetails', methods=['POST'])
+def getinfobyproject_route():
+    data = request.get_json()
+    text = data.get('text', '')
+    real_projects_list = data.get('projects', [])
+    
+    if real_projects_list:
+        for i in real_projects_list:
+            prompt_aim = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке. Представь, что ты лучший в мире смысловик и копирайтер.
+Сделай глубокий вдох и, думая шаг за шагом, выполни задание:
+Опираясь на данный текст, определи цель ЭТОГО проекта: {i}.  Выбирай и анализируй информацию только об ЭТОМ проекте.
+Формат ответа: "Проект: *название проекта*. Цель этого проекта: *здесь текст*".
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Текст: {text}[/INST]''')
+            gk = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке. Представь, что ты лучший в мире смысловик и копирайтер.
+Сделай глубокий вдох и, думая шаг за шагом, ответь на вопрос: Какие преимущества (выгоду) получит группа компаний по итогам этого проекта: {i}.
+Формат ответа: "По факту реализации проекта Группа Компаний Получит: *здесь текст*".
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Скрипт: {text}[/INST]''')
+            budget = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке.
+Найди в тексте информацию о бюджете этого проекта: {i} .
+Формат ответа: "Бюджет проекта: *здесь цифра*". Если бюджет проекта не указан, то отвечай так: "Такой информации нет."
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Скрипт: {text}[/INST]''')
+            ec_ef = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке.
+Поразмышляй об этом тексте. Как ты думаешь, в чем заключается экономический эффект (экономические преимущества) для компании от этого проекта: {i}. Выбирай и анализируй информацию только об ЭТОМ проекте.
+Формат ответа: "Экономический эффект от проекта: *здесь ответ на вопрос*".
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Скрипт: {text}[/INST]''')
+            deadline = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке.
+Найди в тексте дату, когда должен быть реализован ЭТОТ проект: {i}. Выбирай и анализируй информацию только об ЭТОМ проекте: {i}.
+Формат ответа: "Срок реализации: *здесь текст*".
+Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь. Отвечай на русском языке.
+Скрипт: {text}[/INST]''')
+            new_plan = (f'<s>[INST]Ты всегда отвечаешь на РУССКОМ языке! Только на русском языке. Выбирай и анализируй информацию на русском языке только об ЭТОМ проекте: {i} .+\
+ Представь, что ты лучший в мире смысловик и копирайтер. +\
+Сделай глубокий вдох и, думая шаг за шагом, ответь на русском языке на вопрос: Какие действия участники решили предпринять, чтобы отлич��о выполнить проект?+\
+Используй такой формат ответа: "Решения: *текст*". Ты больше ничего не говоришь, не комментируешь, не выражаешь свои мысли, вообще ничего больше не говоришь! Ищи ответ в тексте: {text}. Отвечай на русском языке!!! [/INST]')
+            conclusion = (f'''<s>[INST]Ты всегда отвечаешь на РУССКОМ языке. Поразмышляй о тексте.
+Какой вывод заключили участники относительно ЭТОГО проекта: {i}.
+Используй такой формат ответа: "Вывод: *Тут описывается принятое решение.*." Конец ответа.
+Текст: {text}[/INST]''')
+            p = [prompt_aim, gk, budget, ec_ef, deadline, new_plan, conclusion]
+            real_projects_list['prompts'] = p
+    elif not real_projects_list:
+        return jsonify({'error': 'Проекты не выбраны'})
+    
+    final = []
+    for project in real_projects_list:
+        for prompt in project['prompts']:
+            result = generate(prompt[:52000])
+            if result is not None:
+                final[project] += '\n\n'+result + '\n\n'
+        final[project] = final[project].replace("Конец ответа", "")
+        final[project] = final[project].replace('</s>', '')
+        final[project] = final[project].strip()
+
+    return jsonify({'result': final})
+
+
 if __name__ == '__main__':
     app.run(debug=False, host='0.0.0.0', port=7860)