Update app.py

app.py

@@ -6,11 +6,16 @@ from prophet import Prophet
 import io
 from PIL import Image
 
-# Ссылки на CSV-файлы
+# Первые наборы CSV-файлов
 URL_DASHA = "https://raw.githubusercontent.com/fruitpicker01/Storage_Dasha_2025/main/messages.csv"
-URL_LERA = "https://raw.githubusercontent.com/fruitpicker01/Storage_Lera_2025/main/messages.csv"
+URL_LERA  = "https://raw.githubusercontent.com/fruitpicker01/Storage_Lera_2025/main/messages.csv"
 URL_SVETA = "https://raw.githubusercontent.com/fruitpicker01/Storage_Sveta_2025/main/messages.csv"
 
+# Вторые наборы CSV-файлов
+URL_DASHA_2 = "https://raw.githubusercontent.com/fruitpicker01/Storage_2_Dasha_2025/main/messages.csv"
+URL_LERA_2  = "https://raw.githubusercontent.com/fruitpicker01/Storage_2_Lera_2025/main/messages.csv"
+URL_SVETA_2 = "https://raw.githubusercontent.com/fruitpicker01/Storage_2_Sveta_2025/main/messages.csv"
+
 def read_and_process_data(url, user_name):
     """
     Считывает CSV, отбирает нужные столбцы,
@@ -141,18 +146,47 @@ def make_average_forecast(total_by_date, end_date_str="2025-03-15"):
 def process_data():
     print("\n=== Начинаем process_data (Seaborn + Prophet + средний) ===")
     
-    # 1) Считываем CSV
+    # Чтение основного файла
     dasha_count, dasha_daily = read_and_process_data(URL_DASHA, "Даша")
     lera_count,  lera_daily  = read_and_process_data(URL_LERA, "Лера")
     sveta_count, sveta_daily = read_and_process_data(URL_SVETA, "Света")
     
-    total_count = dasha_count + lera_count + sveta_count
+    # Чтение второго набора данных (с обработкой ошибок)
+    try:
+        dasha_count2, dasha_daily2 = read_and_process_data(URL_DASHA_2, "Даша (2)")
+    except Exception as e:
+        print(f"[Даша (2)] Ошибка при чтении дополнительного CSV: {e}")
+        dasha_count2, dasha_daily2 = 0, pd.DataFrame(columns=["date", "count", "user"])
+    
+    try:
+        lera_count2, lera_daily2 = read_and_process_data(URL_LERA_2, "Лера (2)")
+    except Exception as e:
+        print(f"[Лера (2)] Ошибка при чтении дополнительного CSV: {e}")
+        lera_count2, lera_daily2 = 0, pd.DataFrame(columns=["date", "count", "user"])
+    
+    try:
+        sveta_count2, sveta_daily2 = read_and_process_data(URL_SVETA_2, "Света (2)")
+    except Exception as e:
+        print(f"[Света (2)] Ошибка при чтении дополнительного CSV: {e}")
+        sveta_count2, sveta_daily2 = 0, pd.DataFrame(columns=["date", "count", "user"])
+    
+    # Объединяем основные и дополнительные данные по каждому пользователю
+    dasha_count_total = dasha_count + dasha_count2
+    lera_count_total  = lera_count  + lera_count2
+    sveta_count_total = sveta_count + sveta_count2
+    
+    dasha_daily_total = pd.concat([dasha_daily, dasha_daily2], ignore_index=True)
+    lera_daily_total  = pd.concat([lera_daily, lera_daily2], ignore_index=True)
+    sveta_daily_total = pd.concat([sveta_daily, sveta_daily2], ignore_index=True)
+    
+    total_count = dasha_count_total + lera_count_total + sveta_count_total
     print(f"Суммарное количество (Д+Л+С): {total_count}")
     
-    # 2) Прогресс-бары
-    dasha_percent = round((dasha_count / 234) * 100) if 234 else 0
-    lera_percent = round((lera_count / 234) * 100) if 234 else 0
-    sveta_percent = round((sveta_count / 234) * 100) if 234 else 0
+    # Остальной код (прогресс-бары, объединение DataFrame, графики) 
+    # замените исходные переменные на объединённые *_total
+    dasha_percent = round((dasha_count_total / 234) * 100) if 234 else 0
+    lera_percent  = round((lera_count_total  / 234) * 100) if 234 else 0
+    sveta_percent = round((sveta_count_total / 234) * 100) if 234 else 0
     total_percent = round((total_count / 702) * 100) if 702 else 0
     
     def get_progress_bar(label, abs_val, pct):
@@ -168,14 +202,14 @@ def process_data():
         </div>
         """
     bars_html = (
-        get_progress_bar("Даша", dasha_count, dasha_percent) +
-        get_progress_bar("Лера", lera_count, lera_percent) +
-        get_progress_bar("Света", sveta_count, sveta_percent) +
+        get_progress_bar("Даша", dasha_count_total, dasha_percent) +
+        get_progress_bar("Лера", lera_count_total, lera_percent) +
+        get_progress_bar("Света", sveta_count_total, sveta_percent) +
         get_progress_bar("Всего", total_count, total_percent)
     )
     
-    # 3) Формируем общий DF
-    daily_all = pd.concat([dasha_daily, lera_daily, sveta_daily], ignore_index=True)
+    # Объединение ежедневных данных для построения графика
+    daily_all = pd.concat([dasha_daily_total, lera_daily_total, sveta_daily_total], ignore_index=True)
     daily_all = daily_all.dropna(subset=["date"])
     daily_all = daily_all.sort_values(["user", "date"])
     daily_all["cumulative"] = daily_all.groupby("user")["count"].cumsum()