flask_inference_api_g

Sleeping

App Files Files Community

DmitrMakeev commited on Mar 20

Commit

ed1f354

verified ·

1 Parent(s): d9f70cd

Update app.py

Browse files

Files changed (1) hide show

app.py +17 -34

app.py CHANGED Viewed

@@ -327,17 +327,16 @@ def plot_ph_week():
         # Получаем номер недели из параметра (по умолчанию 1)
         week_number = request.args.get('week', default=1, type=int)
-        # Ограничиваем границы (1-30 недель)
         week_number = max(1, min(30, week_number))
         # Подключаемся к базе данных
         conn = sqlite3.connect('system_data.db')
         cursor = conn.cursor()
-        # Проверяем существование таблицы
         cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='system_data'")
         if not cursor.fetchone():
-            conn.close()
             return '''
             <html>
                 <body>
@@ -348,7 +347,7 @@ def plot_ph_week():
             </html>
             '''
-        # Выборка данных за указанную неделю
         cursor.execute('''
             SELECT dey, ph
             FROM system_data
@@ -359,7 +358,7 @@ def plot_ph_week():
         conn.close()
-        # Если данных нет
         if not rows:
             return f'''
             <html>
@@ -375,42 +374,35 @@ def plot_ph_week():
         # Создаём словарь {день недели: список значений pH}
         data_by_day = {i: [] for i in range(1, 8)}
         for day, ph in rows:
-            # Преобразуем ph в float, если оно строка
-            try:
-                ph_value = float(ph)
-                data_by_day[day].append(ph_value)
-            except (ValueError, TypeError):
-                continue  # Пропускаем некорректные значения
-        # Очищаем предыдущий график
-        plt.clf()
-        # Создаём график
         plt.figure(figsize=(10, 5))
         for day in range(1, 8):
-            if data_by_day[day]:  # Проверяем, есть ли данные
                 plt.scatter([day] * len(data_by_day[day]), data_by_day[day], label=f'День {day}')
         plt.title(f'Значения pH за {week_number}-ю неделю')
         plt.xlabel('День недели')
         plt.ylabel('Значение pH')
-        plt.xticks(range(1, 8))
         plt.grid(True)
-        plt.legend()  # Добавляем легенду для ясности
         # Сохраняем график в буфер
         buffer = io.BytesIO()
         plt.savefig(buffer, format='png')
         buffer.seek(0)
-        # Кодируем в base64
         plot_data = base64.b64encode(buffer.getvalue()).decode('utf-8')
-        # Закрываем буфер и очищаем график
-        buffer.close()
-        plt.close()  # Закрываем фигуру, чтобы избежать утечек памяти
-        # Возвращаем HTML
         return f'''
         <html>
             <body>
@@ -424,16 +416,7 @@ def plot_ph_week():
         '''
     except Exception as e:
-        # Возвращаем HTML вместо JSON для консистентности
-        return f'''
-        <html>
-            <body>
-                <h1>Ошибка!</h1>
-                <p>Произошла ошибка: {str(e)}</p>
-                <button onclick="window.location.href='/plot_ph_week'">Назад</button>
-            </body>
-        </html>
-        ''', 500

         # Получаем номер недели из параметра (по умолчанию 1)
         week_number = request.args.get('week', default=1, type=int)
+        # Проверяем границы (1-30 недель)
         week_number = max(1, min(30, week_number))
         # Подключаемся к базе данных
         conn = sqlite3.connect('system_data.db')
         cursor = conn.cursor()
+        # Проверяем, существует ли таблица
         cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='system_data'")
         if not cursor.fetchone():
             return '''
             <html>
                 <body>
             </html>
             '''
+        # Выбираем данные за указанную неделю
         cursor.execute('''
             SELECT dey, ph
             FROM system_data
         conn.close()
+        # Если данных нет, вывести сообщение
         if not rows:
             return f'''
             <html>
         # Создаём словарь {день недели: список значений pH}
         data_by_day = {i: [] for i in range(1, 8)}
         for day, ph in rows:
+            if ph is not None:  # Проверка, что pH не пустой
+                try:
+                    data_by_day[day].append(float(ph))
+                except ValueError:
+                    print(f"Ошибка приведения pH ({ph}) в день {day}")
+        # Создаём график (X: дни 1-7, Y: pH)
         plt.figure(figsize=(10, 5))
+        # Наносим точки для каждого дня
         for day in range(1, 8):
+            if data_by_day[day]:  # Проверяем, есть ли данные для дня
                 plt.scatter([day] * len(data_by_day[day]), data_by_day[day], label=f'День {day}')
         plt.title(f'Значения pH за {week_number}-ю неделю')
         plt.xlabel('День недели')
         plt.ylabel('Значение pH')
+        plt.xticks(range(1, 8))  # Подписи оси X (1-7)
         plt.grid(True)
         # Сохраняем график в буфер
         buffer = io.BytesIO()
         plt.savefig(buffer, format='png')
         buffer.seek(0)
+        # Кодируем график в base64
         plot_data = base64.b64encode(buffer.getvalue()).decode('utf-8')
+        # Возвращаем HTML с кнопками управления
         return f'''
         <html>
             <body>
         '''
     except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)}), 500