Update app.py

app.py

@@ -6,6 +6,7 @@ import matplotlib.pyplot as plt
 from io import BytesIO
 import tempfile
 import os
+from PIL import Image
 
 # Функция для обработки PDF и извлечения данных
 def process_pdf(file):
@@ -76,7 +77,7 @@ def process_pdf(file):
 
     except Exception as e:
         print(f"Ошибка при обработке PDF-файла: {e}")
-        return pd.DataFrame(), pd.DataFrame(), pd.DataFrame(), None, None, None  # Возвращаем пустые данные в случае ошибки
+        return pd.DataFrame(), pd.DataFrame(), None, None, None, None  # Возвращаем пустые данные в случае ошибки
 
     # Преобразуем список данных в DataFrame
     df = pd.DataFrame(data_list)
@@ -104,7 +105,7 @@ def process_pdf(file):
     df_percentage = df[['№ локальной сметы', 'Наименование']].copy()
     df_percentage['Процент от общей суммы'] = ((df['ВСЕГО ПО СМЕТЕ'].iloc[:-1] / total_sum) * 100).round(2)
 
-    # Создание таблицы структуры расходов по категориям
+    # Создание круговой диаграммы
     category_sums = {
         'Категория': ['Рабочие', 'Техника', 'Материалы', 'Перевозки', 'Оборудование'],
         'Сумма': [
@@ -119,7 +120,40 @@ def process_pdf(file):
     df_category_percentage = pd.DataFrame(category_sums)
     df_category_percentage['Процент от общей суммы'] = ((df_category_percentage['Сумма'] / total_sum) * 100).round(2)
 
-    # Создание словаря сопоставления категорий и цветов (для диаграммы, если понадобится)
+    # Создание словаря сопоставления категорий и цветов
+    color_map = {
+        'Рабочие': '#DCEDFC',
+        'Техника': '#94B4D4',
+        'Материалы': '#3C47D6',
+        'Перевозки': '#94B4D4',
+        'Оборудование': '#0E86D4'
+    }
+
+    # Фильтрация категорий с процентом >=1%
+    filtered_df = df_category_percentage[df_category_percentage['Процент от общей суммы'] >= 1].reset_index(drop=True)
+
+    # Генерация списка цветов в соответствии с порядком категорий
+    colors = [color_map.get(category, '#FFFFFF') for category in filtered_df['Категория']]
+
+    # Сохраняем диаграмму в BytesIO с добавлением пустого пространства
+    fig, ax = plt.subplots(figsize=(7, 7))  # Увеличенный размер фигуры
+    ax.pie(
+        filtered_df['Сумма'],
+        labels=filtered_df['Категория'],
+        autopct='%1.2f%%',
+        startangle=90,
+        colors=colors,  # Используем пользовательские цвета
+        labeldistance=1.1,  # Расстояние до надписей
+        pctdistance=0.8  # Расстояние до процентных надписей
+    )
+    plt.title('')
+    plt.axis('equal')  # Круговая диаграмма круглая по форме
+    plt.tight_layout()  # Оптимизация расположения элементов
+
+    pie_chart = BytesIO()
+    plt.savefig(pie_chart, format='png', bbox_inches='tight')  # bbox_inches='tight' для оптимизации
+    plt.close(fig)
+    pie_chart.seek(0)
 
     # Форматирование чисел с пробелами в качестве разделителей тысяч
     def format_number(x):
@@ -146,8 +180,8 @@ def process_pdf(file):
     df_percentage.to_excel(percentage_report_path, index=False)
     df_category_percentage.to_excel(category_report_path, index=False)
 
-    # Возвращаем форматированные DataFrame и пути к файлам
-    return df_formatted, df_percentage_formatted, df_category_percentage_formatted, detailed_report_path, percentage_report_path, category_report_path
+    # Возвращаем форматированные DataFrame и диаграмму
+    return df_formatted, df_percentage_formatted, df_category_percentage_formatted, pie_chart, detailed_report_path, percentage_report_path, category_report_path
 
 # CSS для блокировки копирования текста
 custom_css = """
@@ -166,7 +200,7 @@ with gr.Blocks(css=custom_css) as demo:
         gr.Markdown("### **AIGineer | ИИ УПРАВЛЕНИЕ СТРОИТЕЛЬСТВОМ**", elem_id="header")
         gr.Markdown("**На текущий момент система считает сметы в ценах от 2023**", elem_id="subheader")
 
-    # Загрузка файла
+    # Загрузка файла (текст выравнивается по центру)
     with gr.Row():
         upload_button = gr.File(label="ЗАГРУЗИТЕ СМЕТУ В PDF", elem_id="upload")
 
@@ -178,50 +212,72 @@ with gr.Blocks(css=custom_css) as demo:
                 "% соотношение локальных смет",
                 "Структура расходов по категориям"
             ],
-            label="Выберите отчет"
+            label="ВЫБЕРИТЕ ОТЧЕТ"
         )
 
+    # Описание и поля отчета
+    description = gr.Markdown(visible=False)
+    preview_table = gr.Dataframe(visible=False)
+
     # Кнопка для получения отчета
     with gr.Row():
         get_report_button = gr.Button("Получить отчет")
 
-    # Заголовок отчета
-    report_title = gr.Markdown(visible=False)
-
     # Отображение отчета и кнопка скачивания
     with gr.Row():
         output_table = gr.Dataframe(visible=False)
-    # Кнопка для скачивания отчета
+        output_image = gr.Image(type="pil", visible=False)
+
     with gr.Row():
         download_excel = gr.File(label="Скачать отчет", visible=False)
 
+    # Функция для обновления описания и полей отчета при выборе
+    def update_preview(selected_report):
+        if selected_report == "Детализированный бюджет":
+            description_text = "Отчет дает полное представление о том, как распределяются средства по каждому разделу сметной документации..."
+            preview_df = pd.DataFrame(columns=["№ локальной сметы", "Наименование", "Рабочие", "Техника", "Материалы", "Перевозки", "Оборудование", "ВСЕГО ПО СМЕТЕ"])
+            return gr.update(value=description_text, visible=True), gr.update(value=preview_df, visible=True)
+        elif selected_report == "% соотношение локальных смет":
+            description_text = "Важный инструмент для стратегического планирования, который помогает легко сравнивать между собой разделы проекта..."
+            preview_df = pd.DataFrame(columns=["№ локальной сметы", "Наименование", "Процент от общей суммы"])
+            return gr.update(value=description_text, visible=True), gr.update(value=preview_df, visible=True)
+        elif selected_report == "Структура расходов по категориям":
+            description_text = "Позволяет увидеть, как распределяются затраты по основным категориям..."
+            return gr.update(value=description_text, visible=True), gr.update(visible=False)
+        else:
+            return gr.update(visible=False), gr.update(visible=False)
+
+    # Связываем выбор отчета с обновлением описания и полей
+    report_selector.change(
+        update_preview,
+        inputs=report_selector,
+        outputs=[description, preview_table]
+    )
+
     # Функция обновления отчета
     def update_report(file, selected_report):
         if not file or not selected_report:
             return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
         # Обрабатываем PDF и получаем данные
-        df_formatted, df_percentage_formatted, df_category_percentage_formatted, detailed_report_path, percentage_report_path, category_report_path = process_pdf(file.name)
+        df_formatted, df_percentage_formatted, df_category_percentage_formatted, pie_chart, detailed_report_path, percentage_report_path, category_report_path = process_pdf(file.name)
 
         if selected_report == "Детализированный бюджет":
-            title = "### ДЕТАЛИЗИРОВАННЫЙ БЮДЖЕТ"
             return (
-                gr.update(value=title, visible=True),
                 gr.update(value=df_formatted, visible=True),
+                gr.update(visible=False),
                 gr.update(value=detailed_report_path, visible=True, label="Скачать детализированный отчет")
             )
         elif selected_report == "% соотношение локальных смет":
-            title = "### ПРОЦЕНТНОЕ РАСПРЕДЕЛЕНИЕ ЛОКАЛЬНЫХ СМЕТ"
             return (
-                gr.update(value=title, visible=True),
                 gr.update(value=df_percentage_formatted, visible=True),
+                gr.update(visible=False),
                 gr.update(value=percentage_report_path, visible=True, label="Скачать процентное соотношение")
             )
         elif selected_report == "Структура расходов по категориям":
-            title = "### СТРУКТУРА РАСХОДОВ ПО КАТЕГОРИЯМ"
             return (
-                gr.update(value=title, visible=True),
-                gr.update(value=df_category_percentage_formatted, visible=True),
+                gr.update(visible=False),
+                gr.update(value=Image.open(pie_chart), visible=True),
                 gr.update(value=category_report_path, visible=True, label="Скачать отчет по категориям")
             )
         else:
@@ -231,7 +287,7 @@ with gr.Blocks(css=custom_css) as demo:
     get_report_button.click(
         update_report,
         inputs=[upload_button, report_selector],
-        outputs=[report_title, output_table, download_excel]
+        outputs=[output_table, output_image, download_excel]
     )
 
 # Запуск приложения