Update app.py

app.py

@@ -7,6 +7,8 @@ from io import BytesIO
 import tempfile
 import os
 from PIL import Image
+import openpyxl
+from openpyxl.styles import Font, PatternFill
 
 # Функция для обработки PDF и извлечения данных
 def process_pdf(file):
@@ -65,7 +67,7 @@ def process_pdf(file):
 
             # Сохраняем данные в список с обновленными названиями столбцов
             data_list.append({
-                '№ локальной сметы': estimate_number,
+                '№ сметы': estimate_number,
                 'Наименование': estimate_name,
                 'Рабочие': labor_cost,
                 'Техника': machine_cost,
@@ -82,12 +84,9 @@ def process_pdf(file):
     # Преобразуем список данных в DataFrame
     df = pd.DataFrame(data_list)
     
-    # Убираем лишние пробелы и символы из названий столбцов
-    df.columns = df.columns.str.strip()
-
     # Рассчитываем общую сумму по всем сметам для каждой категории
     total_row = pd.DataFrame({
-        '№ локальной сметы': ['Итого'],
+        '№ сметы': ['Итого'],
         'Наименование': [''],
         'Рабочие': [df['Рабочие'].sum()],
         'Техника': [df['Техника'].sum()],
@@ -102,7 +101,7 @@ def process_pdf(file):
 
     # Создаем таблицу с процентным соотношением, округленным до двух знаков после запятой
     total_sum = df['ВСЕГО ПО СМЕТЕ'].iloc[:-1].sum()  # исключаем строку "Итого"
-    df_percentage = df[['№ локальной сметы', 'Наименование']].copy()
+    df_percentage = df[['№ сметы', 'Наименование']].copy()
     df_percentage['Процент от общей суммы'] = ((df['ВСЕГО ПО СМЕТЕ'].iloc[:-1] / total_sum) * 100).round(2)
 
     # Создание круговой диаграммы
@@ -142,12 +141,44 @@ def process_pdf(file):
     detailed_report_path = os.path.join(temp_dir, "detailed_report.xlsx")
     percentage_report_path = os.path.join(temp_dir, "percentage_report.xlsx")
 
-    df.to_excel(detailed_report_path, index=False)
+    # Сохраняем DataFrame с настройками стиля
+    save_to_excel_with_styles(df, detailed_report_path)
     df_percentage.to_excel(percentage_report_path, index=False)
 
     # Возвращаем пути к файлам и диаграмму
     return df, df_percentage, pie_chart, detailed_report_path, percentage_report_path
 
+# Функция для сохранения DataFrame в Excel с форматированием
+def save_to_excel_with_styles(df, filepath):
+    writer = pd.ExcelWriter(filepath, engine='openpyxl')
+    df.to_excel(writer, index=False)
+    workbook = writer.book
+    worksheet = writer.sheets['Sheet1']
+
+    # Устанавливаем шрифт Intern размером 12 для всех ячеек
+    intern_font = Font(name='Intern', size=12)
+    for row in worksheet.iter_rows():
+        for cell in row:
+            cell.font = intern_font
+
+    # Устанавливаем ширину столбцов и цвет заливки
+    column_widths = {
+        'A': 16, 'B': 50, 'C': 18, 'D': 18, 'E': 18, 'F': 18, 'G': 18, 'H': 18
+    }
+    fill_colors = {
+        'C': 'DCEDFC', 'D': '94B4D4', 'E': '3866C9',
+        'F': 'DCEDFC', 'G': '94B4D4', 'H': '3866C9'
+    }
+
+    for col, width in column_widths.items():
+        worksheet.column_dimensions[col].width = width
+
+    for col, color in fill_colors.items():
+        for cell in worksheet[col]:
+            cell.fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
+
+    writer.save()
+
 # CSS для блокировки копирования текста
 custom_css = """
 * {
@@ -158,46 +189,30 @@ custom_css = """
 }
 """
 
-# Создаем интерфейс Gradio с блоками и адаптивным дизайном
+# Интерфейс Gradio
 with gr.Blocks(css=custom_css) as demo:
-    # Основной заголовок и подзаголовок
     with gr.Row():
-        gr.Markdown("### **AIGineer | ИИ УПРАВЛЕНИЕ СТРОИТЕЛЬСТВОМ**", elem_id="header")
-        gr.Markdown("**На текущий момент система считает сметы в ценах от 2023**", elem_id="subheader")
+        gr.Markdown("### **AIGineer | ИИ УПРАВЛЕНИЕ СТРОИТЕЛЬСТВОМ**")
+        gr.Markdown("**На текущий момент система считает сметы в ценах от 2023**")
 
-    # Загрузка файла (текст выравнивается по центру)
     with gr.Row():
-        upload_button = gr.File(label="ЗАГРУЗИТЕ СМЕТУ В PDF", elem_id="upload")
+        upload_button = gr.File(label="ЗАГРУЗИТЕ СМЕТУ В PDF")
 
-    # Первая таблица с заголовком и описанием
-    with gr.Row():
-        gr.Markdown("<br>")  # Добавляем расстояние перед таблицей
-    with gr.Row():
-        gr.Markdown("<h3 style='text-align: center;'>ДЕТАЛИЗИРОВАННЫЙ БЮДЖЕТ</h3>", elem_id="title1")
     with gr.Row():
-        gr.Markdown("Отчет дает полное представление о том, как распределяются средства по каждому разделу сметной документации. Он разбивает затраты на ключевые категории — оплата труда, техника, материалы, логистика и оборудование. Это позволяет держать под контролем каждую статью расходов и моментально видеть, где происходят основные затраты.", elem_id="desc1")
+        gr.Markdown("<h3 style='text-align: center;'>ДЕТАЛИЗИРОВАННЫЙ БЮДЖЕТ</h3>")
     with gr.Row():
-        output_table_1 = gr.Dataframe(headers=["№ локальной сметы", "Наименование", "Рабочие", "Техника", "Материалы", "Перевозки", "Оборудование", "ВСЕГО ПО СМЕТЕ"], elem_id="table1")
+        output_table_1 = gr.Dataframe(headers=["№ сметы", "Наименование", "Рабочие", "Техника", "Материалы", "Перевозки", "Оборудование", "ВСЕГО ПО СМЕТЕ"])
 
-    # Расстояние между таблицами
-    with gr.Row():
-        gr.Markdown("<br>")  # Добавляем расстояние между таблицами
-
-    # Вторая таблица с заголовком и описанием
-    with gr.Row():
-        gr.Markdown("<h3 style='text-align: center;'>ПРОЦЕНТНОЕ РАСПРЕДЕЛЕНИЕ ЛОКАЛЬНЫХ СМЕТ</h3>", elem_id="title2")
     with gr.Row():
-        gr.Markdown("Важный инструмент для стратегического планирования, который помогает легко сравнивать между собой разделы проекта, понимая, какие из них требуют большего финансирования, а какие — меньшего. Это позволяет эффективно распределять ресурсы и достигать максимальной отдачи.", elem_id="desc2")
+        gr.Markdown("<h3 style='text-align: center;'>ПРОЦЕНТНОЕ РАСПРЕДЕЛЕНИЕ ЛОКАЛЬНЫХ СМЕТ</h3>")
     with gr.Row():
-        output_table_2 = gr.Dataframe(headers=["№ локальной сметы", "Наименование", "Процент от общей суммы"], elem_id="table2")
+        output_table_2 = gr.Dataframe(headers=["№ сметы", "Наименование", "Процент от общей суммы"])
 
-    # Третья таблица с диаграммой
     with gr.Row():
-        gr.Markdown("<h3 style='text-align: center;'>СТРУКТУРА РАСХОДОВ ПО КАТЕГОРИЯМ</h3>", elem_id="title3")
+        gr.Markdown("<h3 style='text-align: center;'>СТРУКТУРА РАСХОДОВ ПО КАТЕГОРИЯМ</h3>")
     with gr.Row():
-        output_plot = gr.Image(type="pil", elem_id="plot")
+        output_plot = gr.Image(type="pil")
 
-    # Кнопки для скачивания Excel-файлов
     with gr.Row():
         download_excel_1 = gr.File(label="Скачать детализированный отчет")
         download_excel_2 = gr.File(label="Скачать процентное соотношение")