Update app.py

app.py

@@ -8,70 +8,116 @@ from datetime import datetime, timedelta
 import pytz
 import time
 import threading
-from huggingface_hub import HfApi  # импорт API для работы с Hugging Face
+from huggingface_hub import HfApi, hf_hub_download  # Импорт API для работы с Hugging Face
 
-# Подключение к базе данных SQLite3
-conn = sqlite3.connect('auth_system.db', check_same_thread=False)
+# --- Настройки Hugging Face ---
+REPO_ID = "Testbase1/testsett"  # Замените на ваш репозиторий
+DB_FILENAME = "auth_system.db"
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Токен для записи (загрузки)
+HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")  # Токен для чтения (скачивания)
+
+# --- Подключение к базе данных SQLite3 ---
+conn = sqlite3.connect(DB_FILENAME, check_same_thread=False)
 c = conn.cursor()
 
-# Создание таблиц, если они не существуют
-c.execute('''
-CREATE TABLE IF NOT EXISTS users (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    username TEXT NOT NULL UNIQUE,
-    token TEXT NOT NULL
-)
-''')
-conn.commit()
-
-c.execute('''
-CREATE TABLE IF NOT EXISTS products (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    name TEXT NOT NULL,
-    description TEXT,
-    purchase_price REAL NOT NULL,
-    sale_price REAL NOT NULL,
-    quantity_in_stock INTEGER NOT NULL DEFAULT 0,
-    user_id INTEGER NOT NULL,
-    FOREIGN KEY(user_id) REFERENCES users(id)
-)
-''')
-conn.commit()
-
-c.execute('''
-CREATE TABLE IF NOT EXISTS cart (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    product_id INTEGER NOT NULL,
-    quantity INTEGER NOT NULL,
-    UNIQUE(product_id)
-)
-''')
-conn.commit()
-
-c.execute('''
-CREATE TABLE IF NOT EXISTS sales (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    sale_id TEXT NOT NULL,
-    product_id INTEGER NOT NULL,
-    quantity INTEGER NOT NULL,
-    sale_date TIMESTAMP NOT NULL,
-    total_amount REAL NOT NULL,
-    FOREIGN KEY(product_id) REFERENCES products(id)
-)
-''')
-conn.commit()
-
-# Функция для генерации случайного 13-значного токена
+# --- Создание таблиц, если они не существуют ---
+def create_tables():
+    c.execute('''
+    CREATE TABLE IF NOT EXISTS users (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        username TEXT NOT NULL UNIQUE,
+        token TEXT NOT NULL
+    )
+    ''')
+    conn.commit()
+
+    c.execute('''
+    CREATE TABLE IF NOT EXISTS products (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT NOT NULL,
+        description TEXT,
+        purchase_price REAL NOT NULL,
+        sale_price REAL NOT NULL,
+        quantity_in_stock INTEGER NOT NULL DEFAULT 0,
+        user_id INTEGER NOT NULL,
+        FOREIGN KEY(user_id) REFERENCES users(id)
+    )
+    ''')
+    conn.commit()
+
+    c.execute('''
+    CREATE TABLE IF NOT EXISTS cart (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        product_id INTEGER NOT NULL,
+        quantity INTEGER NOT NULL,
+        UNIQUE(product_id)
+    )
+    ''')
+    conn.commit()
+
+    c.execute('''
+    CREATE TABLE IF NOT EXISTS sales (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        sale_id TEXT NOT NULL,
+        product_id INTEGER NOT NULL,
+        quantity INTEGER NOT NULL,
+        sale_date TIMESTAMP NOT NULL,
+        total_amount REAL NOT NULL,
+        FOREIGN KEY(product_id) REFERENCES products(id)
+    )
+    ''')
+    conn.commit()
+
+# --- Функция для скачивания базы данных из Hugging Face ---
+def download_db_from_hf():
+    """Скачивает файл базы данных из репозитория Hugging Face."""
+    try:
+        hf_hub_download(
+            repo_id=REPO_ID,
+            filename=DB_FILENAME,
+            repo_type="dataset",
+            token=HF_TOKEN_READ,
+            local_dir=".",  # Текущая директория
+            local_dir_use_symlinks=False
+        )
+        print("База данных успешно скачана из Hugging Face.")
+        return True
+    except Exception as e:
+        print("Ошибка при скачивании базы данных:", e)
+        return False
+
+# --- Функция для загрузки базы данных в Hugging Face ---
+def upload_db_to_hf():
+    """Загружает файл базы данных в репозиторий Hugging Face."""
+    try:
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=DB_FILENAME,
+            path_in_repo=DB_FILENAME,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=HF_TOKEN_WRITE
+        )
+        print("Резервная копия базы данных успешно загружена на Hugging Face.")
+    except Exception as e:
+        print("Ошибка при загрузке резервной копии:", e)
+
+# --- Функция периодического резервного копирования ---
+def periodic_backup():
+    """Запускает бесконечный цикл резервного копирования с интервалом в 1 час."""
+    while True:
+        upload_db_to_hf()
+        time.sleep(15)  # ожидание 1 час
+
+# --- Вспомогательные функции (генерация токена, получение ID пользователя и т.д.) ---
 def generate_token():
     return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
 
-# Функция для получения идентификатора пользователя по токену
 def get_user_id_by_token(token):
     c.execute("SELECT id FROM users WHERE token=?", (token,))
     user = c.fetchone()
     return user[0] if user else None
 
-# Функция для получения товаров из корзины
 def get_cart_summary(user_id):
     c.execute('''
     SELECT p.name, c.quantity, p.sale_price, c.quantity * p.sale_price
@@ -84,34 +130,32 @@ def get_cart_summary(user_id):
     total_price = sum(item[3] for item in items)
     return items, total_quantity, total_price
 
-# Функция для добавления продажи в отчет
 def record_sales():
     sales_details = []
-    sale_id = ''.join(random.choices(string.ascii_letters + string.digits, k=8))  # Уникальный идентификатор сделки
+    sale_id = ''.join(random.choices(string.ascii_letters + string.digits, k=8))
     sale_date = datetime.now(pytz.timezone('Asia/Bishkek')).strftime('%Y-%m-%d %H:%M:%S')
-    
+
     c.execute('SELECT product_id, quantity FROM cart')
     cart_items = c.fetchall()
-    
+
     total_amount = 0.0
     for product_id, quantity in cart_items:
         c.execute('SELECT sale_price FROM products WHERE id=?', (product_id,))
         sale_price = c.fetchone()[0]
-        
+
         amount = quantity * sale_price
         total_amount += amount
-        
+
         c.execute('''
         INSERT INTO sales (sale_id, product_id, quantity, sale_date, total_amount)
         VALUES (?, ?, ?, ?, ?)
         ''', (sale_id, product_id, quantity, sale_date, amount))
         sales_details.append((product_id, quantity, amount, sale_date))
-        
+
     c.execute('DELETE FROM cart')
     conn.commit()
     return sales_details, sale_id, total_amount
 
-# Функция для генерации отчета за месяц
 def generate_monthly_report(user_id):
     start_date = (datetime.now().replace(day=1)).strftime('%Y-%m-%d')
     end_date = (datetime.now() + timedelta(days=31)).replace(day=1).strftime('%Y-%m-%d')
@@ -124,15 +168,14 @@ def generate_monthly_report(user_id):
     WHERE s.sale_date BETWEEN ? AND ? AND p.user_id = ?
     GROUP BY p.id
     ''', (start_date, end_date, user_id))
-    
+
     sales_data = c.fetchall()
 
     total_sales = sum(item[3] for item in sales_data)
     total_profit = sum(item[4] for item in sales_data)
-    
+
     return sales_data, total_sales, total_profit
 
-# Функция для получения всех сделок
 def get_all_sales(user_id):
     c.execute('''
     SELECT DISTINCT sale_id, sale_date, SUM(total_amount) AS total_amount
@@ -145,7 +188,6 @@ def get_all_sales(user_id):
     sales_data = c.fetchall()
     return sales_data
 
-# Функция для получения всех подробностей сделок
 def get_sale_details(sale_id):
     c.execute('''
     SELECT p.name, s.quantity, p.sale_price, s.sale_date
@@ -156,7 +198,7 @@ def get_sale_details(sale_id):
     sale_details = c.fetchall()
     return sale_details
 
-# Страница регистрации
+# --- Страницы приложения ---
 def register():
     st.title('Регистрация')
     username = st.text_input("Введите ваше имя пользователя")
@@ -174,7 +216,6 @@ def register():
         else:
             st.error("Неверный пароль администратора!")
 
-# Страница авторизации
 def login():
     st.title('Авторизация')
     token_input = st.text_input("Введите ваш токен")
@@ -182,14 +223,13 @@ def login():
     if st.button("Войти"):
         user_id = get_user_id_by_token(token_input)
         if user_id:
-            st.session_state.logged_in = True  # Устанавливаем состояние авторизации
+            st.session_state.logged_in = True
             st.session_state.username = token_input
             st.session_state.user_id = user_id
             st.success("Добро пожаловать!")
         else:
             st.error("Неверный токен!")
 
-# Форма добавления товара
 def add_product():
     st.title("Добавление товара")
     product_name = st.text_input("Название товара").strip().lower()
@@ -207,23 +247,22 @@ def add_product():
         else:
             st.error("Пожалуйста, введите все обязательные данные.")
 
-# Форма редактирования и удаления товара
 def edit_products():
     st.title("Редактирование товара")
     products = c.execute("SELECT id, name, description, purchase_price, sale_price, quantity_in_stock FROM products WHERE user_id=?", 
                          (st.session_state.user_id,)).fetchall()
-    
+
     if products:
         product_names = [p[1] for p in products]
         product_id = st.selectbox("Выберите товар", product_names)
         product = next(p for p in products if p[1] == product_id)
-        
+
         new_name = st.text_input("Новое название товара", product[1])
         new_description = st.text_area("Новое описание товара", product[2])
         new_purchase_price = st.number_input("Новая приходная цена", min_value=0.0, step=0.01, value=product[3])
         new_sale_price = st.number_input("Новая отпускная цена", min_value=0.0, step=0.01, value=product[4])
         new_quantity_in_stock = st.number_input("Новое количество на складе", min_value=0, step=1, value=product[5])
-        
+
         if st.button("Сохранить изменения"):
             c.execute('''
             UPDATE products
@@ -238,7 +277,6 @@ def edit_products():
             conn.commit()
             st.success("Товар успешно удален!")
 
-# Форма отпуска товара
 def add_to_cart():
     st.title("Отпуск товара")
     products = c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=?", 
@@ -259,10 +297,10 @@ def add_to_cart():
 
             with cols[1]:
                 quantity = st.number_input(f"Количество для '{product[1]}'", min_value=0, max_value=product[3], key=f"quantity_{product[0]}")
-                
+
             with cols[2]:
                 add_button = st.button(f"Добавить в корзину", key=f"add_{product[0]}")
-                
+
                 if add_button:
                     if quantity <= product[3]:
                         c.execute('''
@@ -280,7 +318,7 @@ def add_to_cart():
 
             with cols[3]:
                 remove_button = st.button(f"Удалить из корзины", key=f"remove_{product[0]}")
-                
+
                 if remove_button:
                     if quantity > 0:
                         current_quantity = c.execute("SELECT quantity FROM cart WHERE product_id=?", (product[0],)).fetchone()
@@ -306,13 +344,13 @@ def add_to_cart():
 
         st.subheader("Состояние корзины")
         items, total_quantity, total_price = get_cart_summary(st.session_state.user_id)
-        
+
         if items:
             df = pd.DataFrame(items, columns=["Название", "Количество", "Цена за единицу", "Итого"])
             st.dataframe(df.style.format({"Цена за единицу": "{:.2f}", "Итого": "{:.2f}"}), use_container_width=True)
-            
+
             st.write(f"Общее количество: {total_quantity}, Общая стоимость: {total_price:.2f}")
-            
+
             if st.button("Пробить"):
                 sales_details, sale_id, total_amount = record_sales()
                 if sales_details:
@@ -321,7 +359,7 @@ def add_to_cart():
                     st.dataframe(sale_details_df.style.format({"Сумма": "{:.2f}", "Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
                     st.write(f"**Общая сумма сделки:** {total_amount:.2f}")
                     st.success("Корзина успешно пробита! Все товары добавлены в отчет и корзина очищена.")
-                
+
             if st.button("Сделки"):
                 sales_data = get_all_sales(st.session_state.user_id)
                 if sales_data:
@@ -338,7 +376,6 @@ def add_to_cart():
         else:
             st.info("Корзина пуста.")
 
-# Страница отчета о продажах за месяц
 def monthly_report():
     st.title("Отчет о продажах за месяц")
     sales_data, total_sales, total_profit = generate_monthly_report(st.session_state.user_id)
@@ -346,10 +383,10 @@ def monthly_report():
         df = pd.DataFrame(sales_data, columns=["Название", "Общее количество", "Отпускная цена", "Общие продажи", "Прибыль"])
         st.write(f"**Отчет за {datetime.now().strftime('%B %Y')}**")
         st.dataframe(df.style.format({"Отпускная цена": "{:.2f}", "Общие продажи": "{:.2f}", "Прибыль": "{:.2f}"}), use_container_width=True)
-        
+
         st.write(f"**Общая сумма продаж**: {total_sales:.2f}")
         st.write(f"**Общая сумма прибыли**: {total_profit:.2f}")
-        
+
         if st.button("Сделки"):
             sales_data = get_all_sales(st.session_state.user_id)
             if sales_data:
@@ -366,32 +403,30 @@ def monthly_report():
     else:
         st.info("Нет данных о продажах за этот месяц.")
 
-# --- Функционал резервного копирования базы данных на Hugging Face ---
-hf_token = os.getenv("HF_TOKEN")  # Замените на ваш Hugging Face токен
+# --- Главная функция приложения ---
+def main():
+    # 1. Проверяем, существует ли локальная база данных.
+    if not os.path.exists(DB_FILENAME):
+        st.warning("Локальная база данных не найдена.  Попытка загрузки из Hugging Face...")
+        # 2. Пытаемся скачать базу данных из Hugging Face.
+        if download_db_from_hf():
+            st.success("База данных успешно загружена из Hugging Face!")
+        else:
+            st.error("Не удалось загрузить базу данных из Hugging Face.  Будет создана новая пустая база данных.")
+            create_tables() # Создаем таблицы, если не удалось скачать
+
+    else:
+        st.info("Локальная база данных найдена.")
 
-def upload_db_to_hf():
-    """Загружает файл базы данных в репозиторий Hugging Face (repo_id: Testbase1/testsett)."""
+    # Инициализация таблиц, если база данных только что создана или скачана без таблиц
     try:
-        api = HfApi()
-        api.upload_file(
-            path_or_fileobj="auth_system.db",
-            path_in_repo="auth_system.db",
-            repo_id="Testbase1/testsett",
-            repo_type="dataset",
-            token=hf_token
-        )
-        print("Резервная копия базы данных успешно загружена на Hugging Face.")
-    except Exception as e:
-        print("Ошибка при загрузке резервной копии:", e)
+        c.execute("SELECT * FROM users LIMIT 1")  # Проверяем, существует ли таблица users
+    except sqlite3.OperationalError:
+        st.info("Таблицы не найдены в базе данных. Создание таблиц...")
+        create_tables()  # Создаем таблицы, если они не существуют
+        st.success("Таблицы успешно созданы!")
 
-def periodic_backup():
-    """Запускает бесконечный цикл резервного копирования с интервалом в 1 час."""
-    while True:
-        upload_db_to_hf()
-        time.sleep(3600)  # ожидание 1 час
 
-# --- Главная функция приложения ---
-def main():
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False
 
@@ -421,9 +456,10 @@ def main():
             login()
 
 if __name__ == "__main__":
-    # Глобально запускаем поток резервного копирования, если он ещё не запущен.
+    # 3. Запускаем поток резервного копирования (если он еще не запущен).
     if not globals().get("backup_thread_started", False):
         backup_thread = threading.Thread(target=periodic_backup, daemon=True)
         backup_thread.start()
         globals()["backup_thread_started"] = True
-    main()
+
+    main()