Update app.py

app.py

@@ -152,22 +152,7 @@ def get_sale_details(sale_id):
     sale_details = c.fetchall()
     return sale_details
 
-# Страница авторизации
-def login():
-    st.title('Авторизация')
-    token_input = st.text_input("Введите ваш токен")
-
-    if st.button("Войти"):
-        user_id = get_user_id_by_token(token_input)
-        if user_id:
-            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 register():
     st.title('Регистрация')
     username = st.text_input("Введите ваше имя пользователя")
@@ -185,6 +170,20 @@ def register():
         else:
             st.error("Неверный пароль администратора!")
 
+# Форма авторизации
+def login():
+    st.title('Авторизация')
+    token_input = st.text_input("Введите ваш токен")
+
+    if st.button("Войти"):
+        user_id = get_user_id_by_token(token_input)
+        if user_id:
+            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():
@@ -235,17 +234,11 @@ def edit_products():
             conn.commit()
             st.success("Товар успешно удален!")
 
-# Функция для отпуска товара с поиском
+# Форма отпуска товара
 def add_to_cart():
     st.title("Отпуск товара")
-    
-    # Поисковая строка
-    search_query = st.text_input("Поиск товара", "")
-    
-    # Получаем товары, которые принадлежат пользователю
-    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()
+    products = c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=?", 
+                         (st.session_state.user_id,)).fetchall()
     
     if products:
         for product in products:
@@ -327,55 +320,84 @@ 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("Корзина пуста.")
+
+# Страница отчета о продажах за месяц
+def monthly_report():
+    st.title("Отчет о продажах за месяц")
+    sales_data, total_sales, total_profit = generate_monthly_report()
+    if sales_data:
+        # Создание DataFrame для отчета
+        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()
+            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("Нет товаров, соответствующих вашему запросу.")
+        st.info("Нет данных о продажах за этот месяц.")
 
-# Главная функция
+# Главная функция приложения
 def main():
-    st.sidebar.title("Навигация")
-    page = st.sidebar.selectbox("Выберите страницу", ["Авторизация", "Регистрация", "Добавить товар", "Редактировать товар", "Отпуск товара", "Отчеты"])
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
 
-    # Обрабатываем страницы
-    if page == "Авторизация":
-        login()
-    elif page == "Регистрация":
-        register()
-    elif page == "Добавить товар":
-        if "logged_in" in st.session_state and st.session_state.logged_in:
+    if st.session_state.logged_in:
+        st.sidebar.title(f"Привет, {st.session_state.username}!")
+        option = st.sidebar.selectbox("Выберите действие", ["Добавить товар", "Отпуск товара", "Редактировать товары", "Отчет за месяц", "Выйти"])
+
+        if option == "Добавить товар":
             add_product()
-        else:
-            st.error("Пожалуйста, войдите в систему.")
-    elif page == "Редактировать товар":
-        if "logged_in" in st.session_state and st.session_state.logged_in:
-            edit_products()
-        else:
-            st.error("Пожалуйста, войдите в систему.")
-    elif page == "Отпуск товара":
-        if "logged_in" in st.session_state and st.session_state.logged_in:
+        elif option == "Отпуск товара":
             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("Нет данных для отчета.")
+        elif option == "Редактировать товары":
+            edit_products()
+        elif option == "Отчет за месяц":
+            monthly_report()
+        elif option == "Выйти":
+            st.session_state.logged_in = False
+            st.session_state.username = None
+            st.session_state.user_id = None
+            st.success("Вы вышли из системы!")
+    else:
+        # Показ только страницы авторизации на главной странице
+        login()
+
+        # Добавляем ссылку на регистрацию в сайдбар
+        if st.sidebar.button("Регистрация"):
+            register()
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()