Update app.py

app.py

@@ -185,6 +185,7 @@ def register():
         else:
             st.error("Неверный пароль администратора!")
 
+
 # Форма добавления товара
 def add_product():
     st.title("Добавление товара")
@@ -234,22 +235,17 @@ def edit_products():
             conn.commit()
             st.success("Товар успешно удален!")
 
-    else:
-        st.info("Нет доступных товаров для редактирования или удаления.")
-
-# Форма отпуска товара с поиском
+# Функция для отпуска товара с поиском
 def add_to_cart():
     st.title("Отпуск товара")
     
-    search_query = st.text_input("Поиск товара по названию").strip().lower()  # Поле поиска
+    # Поисковая строка
+    search_query = st.text_input("Поиск товара", "")
     
-    # Получаем все товары пользователя
-    products = c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=?", 
-                         (st.session_state.user_id,)).fetchall()
-
-    # Фильтруем товары по поисковому запросу
-    if search_query:
-        products = [p for p in products if search_query in p[1].lower()]
+    # Получаем товары, которые принадлежат пользователю
+    c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=? AND name LIKE ?", 
+              (st.session_state.user_id, f'%{search_query}%'))
+    products = c.fetchall()
     
     if products:
         for product in products:
@@ -331,57 +327,55 @@ def add_to_cart():
                     st.write(f"**Общая сумма сделки:** {total_amount:.2f}")
                     st.success("Корзина успешно пробита! Все товары добавлены в отчет и корзина очищена.")
                 
-            # Кнопка для отображения сделок
-            if st.button("Сделки"):
-                sales_data = get_all_sales()
-                if sales_data:
-                    sale_ids = [sale[0] for sale in sales_data]
-                    for sale_id, sale_date, total_amount in sales_data:
-                        st.write(f"**Сделка ID:** {sale_id} ({sale_date}) - **Общая сумма:** {total_amount:.2f}")
-                        sale_details = get_sale_details(sale_id)
-                        if sale_details:
-                            df_sales = pd.DataFrame(sale_details, columns=["Название товара", "Количество", "Цена за единицу", "Дата и время"])
-                            st.dataframe(df_sales.style.format({"Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
-                        else:
-                            st.info("Нет деталей для отображения.")
-                else:
-                    st.info("Нет сделок для отображения.")
         else:
             st.info("Корзина пуста.")
+    else:
+        st.info("Нет товаров, соответствующих вашему запросу.")
 
-# Главная функция приложения
+# Главная функция
 def main():
-    st.sidebar.title("Меню")
-    menu = ["Авторизация", "Регистрация", "Добавить товар", "Отпуск товара", "Редактировать товары", "Отчет за месяц"]
-    choice = st.sidebar.selectbox("Выберите действие", menu)
-    
-    if "logged_in" not in st.session_state:
-        st.session_state.logged_in = False
-    
-    if not st.session_state.logged_in:
-        if choice == "Авторизация":
-            login()
-        elif choice == "Регистрация":
-            register()
-    else:
-        st.sidebar.write(f"Пользователь: {st.session_state.username}")
-        
-        if choice == "Добавить товар":
+    st.sidebar.title("Навигация")
+    page = st.sidebar.selectbox("Выберите страницу", ["Авторизация", "Регистрация", "Добавить товар", "Редактировать товар", "Отпуск товара", "Отчеты"])
+
+    # Обрабатываем страницы
+    if page == "Авторизация":
+        login()
+    elif page == "Регистрация":
+        register()
+    elif page == "Добавить товар":
+        if "logged_in" in st.session_state and st.session_state.logged_in:
             add_product()
-        elif choice == "Отпуск товара":
-            add_to_cart()
-        elif choice == "Редактировать товары":
+        else:
+            st.error("Пожалуйста, войдите в систему.")
+    elif page == "Редактировать товар":
+        if "logged_in" in st.session_state and st.session_state.logged_in:
             edit_products()
-        elif choice == "Отчет за месяц":
-            sales_data, total_sales, total_profit = generate_monthly_report()
-            if sales_data:
-                df = pd.DataFrame(sales_data, columns=["Название", "Общее количество", "Цена за единицу", "Общие продажи", "Прибыль"])
-                st.dataframe(df.style.format({"Цена за единицу": "{:.2f}", "Общие продажи": "{:.2f}", "Прибыль": "{:.2f}"}), use_container_width=True)
-                st.write(f"**Общая сумма продаж за месяц:** {total_sales:.2f}")
-                st.write(f"**Общая прибыль за месяц:** {total_profit:.2f}")
-            else:
-                st.info("Нет продаж за текущий месяц.")
+        else:
+            st.error("Пожалуйста, войдите в систему.")
+    elif page == "Отпуск товара":
+        if "logged_in" in st.session_state and st.session_state.logged_in:
+            add_to_cart()
+        else:
+            st.error("Пожалуйста, войдите в систему.")
+    elif page == "Отчеты":
+        if "logged_in" in st.session_state and st.session_state.logged_in:
+            st.title("Отчеты")
+            if st.button("Показать все продажи"):
+                all_sales = get_all_sales()
+                if all_sales:
+                    sales_df = pd.DataFrame(all_sales, columns=["ID продажи", "Дата продажи", "Сумма сделки"])
+                    st.dataframe(sales_df.style.format({"Сумма сделки": "{:.2f}", "Дата продажи": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
+                else:
+                    st.info("Нет продаж.")
+            if st.button("Создать месячный отчет"):
+                sales_data, total_sales, total_profit = generate_monthly_report()
+                if sales_data:
+                    report_df = pd.DataFrame(sales_data, columns=["Название товара", "Общее количество", "Цена за единицу", "Общая сумма продаж", "Прибыль"])
+                    st.dataframe(report_df.style.format({"Цена за единицу": "{:.2f}", "Общая сумма продаж": "{:.2f}", "Прибыль": "{:.2f}"}), use_container_width=True)
+                    st.write(f"**Общая сумма продаж:** {total_sales:.2f}")
+                    st.write(f"**Общая прибыль:** {total_profit:.2f}")
+                else:
+                    st.info("Нет данных для отчета.")
 
-# Запуск приложения
 if __name__ == '__main__':
     main()