Update app.py

app.py

@@ -1,30 +1,40 @@
 import gradio as gr
 import os
+import sqlite3
 
-# Загрузка данных с диска, если они есть
-try:
-    with open("data.txt", "r") as f:
-        data = f.read()
-except FileNotFoundError:
-    data = ""
+# Подключение к базе данных (создаст базу данных, если её нет)
+conn = sqlite3.connect('mydata.db')
+cursor = conn.cursor()
+
+# Создание таблицы, если её нет
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS data (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        content TEXT NOT NULL
+    )
+''')
+
+# Загрузка данных из базы данных
+cursor.execute("SELECT content FROM data ORDER BY id DESC LIMIT 1")
+result = cursor.fetchone()
+data = result[0] if result else ""
 
 def get_data(password):
-    """Функция для получения данных с диска."""
+    """Функция для получения данных из базы данных."""
     if password == os.environ.get("password"):
         return data
     else:
         raise gr.Error("Неверный пароль!")
 
 def send_data(password, text):
-    """Функция для сохранения данных на диск."""
+    """Функция для сохранения данных в базе данных."""
     if password == os.environ.get("password"):
         global data
         data = text
-        with open("data.txt", "w") as f:
-            f.write(data)
+        cursor.execute("INSERT INTO data (content) VALUES (?)", (data,))
+        conn.commit()
         raise gr.Info("Данные успешно сохранены!")
-        with open("data.txt", "r") as l:
-            return l.read()
+        return data
     else:
         raise gr.Error("Неверный пароль!")
 
@@ -46,4 +56,7 @@ with gr.Blocks(css=css) as demo:
     get_button.click(get_data, inputs=password_input, outputs=output_text)
     send_button.click(send_data, inputs=[password_input, text_input], outputs=output_text)
 
-demo.launch()
+demo.launch()
+
+# Закрытие соединения с базой данных при завершении работы
+conn.close()