flask_inference_api_g

Running

App Files Files Community

DmitrMakeev commited on Mar 20

Commit

d9f70cd

verified ·

1 Parent(s): 6a82b1a

Update app.py

Browse files

Files changed (1) hide show

app.py +37 -20

app.py CHANGED Viewed

@@ -327,21 +327,17 @@ def plot_ph_week():
         # Получаем номер недели из параметра (по умолчанию 1)
         week_number = request.args.get('week', default=1, type=int)
-        # Проверяем границы (1-30 недель)
-        if week_number < 1:
-            week_number = 1
-        elif week_number > 30:
-            week_number = 30
         # Подключаемся к базе данных
         conn = sqlite3.connect('system_data.db')
         cursor = conn.cursor()
-        # Проверяем, существует ли таблица
         cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='system_data'")
-        table_exists = cursor.fetchone()
-        if not table_exists:
             return '''
             <html>
                 <body>
@@ -352,7 +348,7 @@ def plot_ph_week():
             </html>
             '''
-        # Выбираем все данные за указанную неделю
         cursor.execute('''
             SELECT dey, ph
             FROM system_data
@@ -363,7 +359,7 @@ def plot_ph_week():
         conn.close()
-        # Если данных нет, вывести сообщение и оставить кнопки
         if not rows:
             return f'''
             <html>
@@ -379,30 +375,42 @@ def plot_ph_week():
         # Создаём словарь {день недели: список значений pH}
         data_by_day = {i: [] for i in range(1, 8)}
         for day, ph in rows:
-            data_by_day[day].append(float(ph))
-        # Создаём график (X: дни 1-7, Y: pH)
-        plt.figure(figsize=(10, 5))
-        # Наносим точки для каждого дня
         for day in range(1, 8):
-            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>
@@ -416,7 +424,16 @@ def plot_ph_week():
         '''
     except Exception as e:
-        return jsonify({'status': 'error', 'message': str(e)}), 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():
+            conn.close()
             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:
+            # Преобразуем 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>
         '''
     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