Cambio de indentacion y agregar visualizaciones

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import sqlite3
+import matplotlib.pyplot as plt
 from dataBaseSetup import create_connection
 
 # Funciones de conexión a la base de datos
@@ -93,6 +94,10 @@ def app():
         if st.button("Execute Query 1", key='3'):
             df = get_stocks_by_category_store(category_name_1, store_name_1)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='category_name', y='total_stock', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 2: Get Order Items":
         st.write("### Query 2: Order Items by Category and Store")
@@ -101,6 +106,10 @@ def app():
         if st.button("Execute Query 2", key='6'):
             df = get_order_items_by_category_store(category_name_2, store_name_2)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='category_name', y='total_items', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 3: Total Sales in Santa Cruz Bikes":
         st.write("### Query 3: Total Sales in Santa Cruz Bikes")
@@ -108,6 +117,10 @@ def app():
         if st.button("Execute Query 3", key='8'):
             df = get_total_sales_by_store_year_month("Santa Cruz Bikes", year_month_3)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='year_month', y='total_sales', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 4: Total Sales in Baldwin Bikes":
         st.write("### Query 4: Total Sales in Baldwin Bikes")
@@ -115,6 +128,10 @@ def app():
         if st.button("Execute Query 4", key='10'):
             df = get_total_sales_by_store_year_month("Baldwin Bikes", year_month_4)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='year_month', y='total_sales', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 5: Total Sales in Rowlett Bikes":
         st.write("### Query 5: Total Sales in Rowlett Bikes")
@@ -122,18 +139,30 @@ def app():
         if st.button("Execute Query 5", key='12'):
             df = get_total_sales_by_store_year_month("Rowlett Bikes", year_month_5)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='year_month', y='total_sales', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 6: Staff with the Highest Number of Orders":
         st.write("### Query 6: Staff with the Highest Number of Orders")
         if st.button("Execute Query 6", key='13'):
             df = get_staff_order_counts(desc=True)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='staff_name', y='order_count', ax=ax)
+                st.pyplot(fig)
 
     elif selected_query == "Query 7: Staff with the Lowest Number of Orders":
         st.write("### Query 7: Staff with the Lowest Number of Orders")
         if st.button("Execute Query 7", key='14'):
             df = get_staff_order_counts(desc=False)
             st.write(df)
+            if not df.empty:
+                fig, ax = plt.subplots()
+                df.plot(kind='bar', x='staff_name', y='order_count', ax=ax)
+                st.pyplot(fig)
 
 if __name__ == "__main__":
     app()

dataBaseSetup.py

@@ -16,15 +16,17 @@ def create_table(conn, create_table_sql):
     try:
         c = conn.cursor()
         c.execute(create_table_sql)
-        print(f"Table created successfully or already exists.")
+        conn.commit()
+        print("Table created successfully or already exists.")
     except Exception as e:
-        print(f"Error creating table: {e}")
+        print("Error creating table:", e)
 
 def import_data_to_table(csv_file, table_name, conn):
     """ Load data from a CSV file and insert it into the specified table """
     try:
         df = pd.read_csv(csv_file)
-        df.to_sql(table_name, conn, if_exists='append', index=False)
+        print(df)
+        df.to_sql(table_name, conn, if_exists='replace', index=False)
         print(f"Data imported successfully into {table_name}.")
     except Exception as e:
         print(f"Error importing data into {table_name}: {e}")
@@ -32,120 +34,119 @@ def import_data_to_table(csv_file, table_name, conn):
 def main():
     # Create a database connection
     conn = create_connection()
+    print(conn)
     if conn is not None:
-    # SQL table creation statements
+        # SQL table creation statements
         tables_sql = {
-        "customers": """ CREATE TABLE IF NOT EXISTS customers (
-                            customer_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            first_name TEXT NOT NULL,
-                            last_name TEXT NOT NULL,
-                            phone TEXT,
-                            email TEXT,
-                            street TEXT,
-                            city TEXT,
-                            state TEXT,
-                            zip_code TEXT
-                        ); """,
-        "staffs": """ CREATE TABLE IF NOT EXISTS staffs (
-                            staff_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            first_name TEXT NOT NULL,
-                            last_name TEXT NOT NULL,
-                            email TEXT,
-                            phone TEXT,
-                            active INTEGER,
-                            store_id INTEGER,
-                            manager_id INTEGER,
-                            FOREIGN KEY (store_id) REFERENCES stores(store_id),
-                            FOREIGN KEY (manager_id) REFERENCES staffs(staff_id)
-                        ); """,
-        "stores": """ CREATE TABLE IF NOT EXISTS stores (
-                            store_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            store_name TEXT NOT NULL,
-                            phone TEXT,
-                            email TEXT,
-                            street TEXT,
-                            city TEXT,
-                            state TEXT,
-                            zip_code TEXT
-                        ); """,
-        "categories": """ CREATE TABLE IF NOT EXISTS categories (
-                            category_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            category_name TEXT NOT NULL
+            "customers": """ CREATE TABLE IF NOT EXISTS customers (
+                                customer_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                first_name TEXT NOT NULL,
+                                last_name TEXT NOT NULL,
+                                phone TEXT,
+                                email TEXT,
+                                street TEXT,
+                                city TEXT,
+                                state TEXT,
+                                zip_code INTEGER
                             ); """,
-        "products": """ CREATE TABLE IF NOT EXISTS products (
-                            product_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            product_name TEXT NOT NULL,
-                            category_id INTEGER,
-                            brand_id INTEGER,
-                            model_year INTEGER,
-                            list_price REAL,
-                            FOREIGN KEY (category_id) REFERENCES categories(category_id),
-                            FOREIGN KEY (brand_id) REFERENCES brands(brand_id)
-                        ); """,
-        "brands": """ CREATE TABLE IF NOT EXISTS brands (
-                            brand_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            brand_name TEXT NOT NULL
-                        ); """,
-        "stocks": """ CREATE TABLE IF NOT EXISTS stocks (
-                            store_id INTEGER,
-                            product_id INTEGER,
-                            quantity INTEGER,
-                            PRIMARY KEY (store_id, product_id),
-                            FOREIGN KEY (store_id) REFERENCES stores(store_id),
-                            FOREIGN KEY (product_id) REFERENCES products(product_id)
-                        ); """,
-        "orders": """ CREATE TABLE IF NOT EXISTS orders (
-                            order_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            customer_id INTEGER,
-                            order_status TEXT,
-                            order_date TEXT,
-                            required_date TEXT,
-                            shipped_date TEXT,
-                            store_id INTEGER,
-                            staff_id INTEGER,
-                            FOREIGN KEY (customer_id) REFERENCES customers(customer_id),
-                            FOREIGN KEY (store_id) REFERENCES stores(store_id),
-                            FOREIGN KEY (staff_id) REFERENCES staffs(staff_id)
-                        ); """,
-        "order_items": """ CREATE TABLE IF NOT EXISTS order_items (
-                            order_id INTEGER,
-                            item_id INTEGER PRIMARY KEY AUTOINCREMENT,
-                            product_id INTEGER,
-                            quantity INTEGER,
-                            list_price REAL,
-                            discount REAL,
-                            FOREIGN KEY (order_id) REFERENCES orders(order_id),
-                            FOREIGN KEY (product_id) REFERENCES products(product_id)
-                            ); """
+            "staffs": """ CREATE TABLE IF NOT EXISTS staffs (
+                                staff_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                first_name TEXT NOT NULL,
+                                last_name TEXT NOT NULL,
+                                email TEXT,
+                                phone TEXT,
+                                active INTEGER,
+                                store_id INTEGER,
+                                manager_id INTEGER,
+                                FOREIGN KEY (store_id) REFERENCES stores(store_id),
+                                FOREIGN KEY (manager_id) REFERENCES staffs(staff_id)
+                            ); """,
+            "stores": """ CREATE TABLE IF NOT EXISTS stores (
+                                store_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                store_name TEXT NOT NULL,
+                                phone TEXT,
+                                email TEXT,
+                                street TEXT,
+                                city TEXT,
+                                state TEXT,
+                                zip_code TEXT
+                            ); """,
+            "categories": """ CREATE TABLE IF NOT EXISTS categories (
+                                category_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                category_name TEXT NOT NULL
+                                ); """,
+            "products": """ CREATE TABLE IF NOT EXISTS products (
+                                product_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                product_name TEXT NOT NULL,
+                                category_id INTEGER,
+                                brand_id INTEGER,
+                                model_year INTEGER,
+                                list_price REAL,
+                                FOREIGN KEY (category_id) REFERENCES categories(category_id),
+                                FOREIGN KEY (brand_id) REFERENCES brands(brand_id)
+                            ); """,
+            "brands": """ CREATE TABLE IF NOT EXISTS brands (
+                                brand_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                brand_name TEXT NOT NULL
+                            ); """,
+            "stocks": """ CREATE TABLE IF NOT EXISTS stocks (
+                                store_id INTEGER,
+                                product_id INTEGER,
+                                quantity INTEGER,
+                                PRIMARY KEY (store_id, product_id),
+                                FOREIGN KEY (store_id) REFERENCES stores(store_id),
+                                FOREIGN KEY (product_id) REFERENCES products(product_id)
+                            ); """,
+            "orders": """ CREATE TABLE IF NOT EXISTS orders (
+                                order_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                customer_id INTEGER,
+                                order_status TEXT,
+                                order_date TEXT,
+                                required_date TEXT,
+                                shipped_date TEXT,
+                                store_id INTEGER,
+                                staff_id INTEGER,
+                                FOREIGN KEY (customer_id) REFERENCES customers(customer_id),
+                                FOREIGN KEY (store_id) REFERENCES stores(store_id),
+                                FOREIGN KEY (staff_id) REFERENCES staffs(staff_id)
+                            ); """,
+            "order_items": """ CREATE TABLE IF NOT EXISTS order_items (
+                                order_id INTEGER,
+                                item_id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                product_id INTEGER,
+                                quantity INTEGER,
+                                list_price REAL,
+                                discount REAL,
+                                FOREIGN KEY (order_id) REFERENCES orders(order_id),
+                                FOREIGN KEY (product_id) REFERENCES products(product_id)
+                                ); """
         }
 
-    # Create tables
-    for table_name, sql_command in tables_sql.items():
-        create_table(conn, sql_command)
+        # Create tables
+        for table_name, sql_command in tables_sql.items():
+            create_table(conn, sql_command)
 
-    # Data import paths
-    data_paths = {
-        "customers": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/customers.csv",
-        "staffs": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/staffs.csv",
-        "products": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/products.csv",
-        "categories": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/categories.csv",
-        "stores": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/stores.csv",
-        "brands": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/brands.csv",
-        "stocks": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/stocks.csv",
-        "orders": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/orders.csv",
-        "order_items": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/order_items.csv",
-    }
+        # Data import paths
+        data_paths = {
+            "customers": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/customers.csv",
+            "staffs": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/staffs.csv",
+            "products": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/products.csv",
+            "categories": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/categories.csv",
+            "stores": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/stores.csv",
+            "brands": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/brands.csv",
+            "stocks": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/stocks.csv",
+            "orders": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/orders.csv",
+            "order_items": "/Users/victorg/Documents/Streamlitapp/InteractiveDBApp/CSV/order_items.csv",
+        }
 
-    # Import data to tables
-    for table_name, csv_path in data_paths.items():
-        import_data_to_table(csv_path, table_name, conn)
+        # Import data to tables
+        for table_name, csv_path in data_paths.items():
+            import_data_to_table(csv_path, table_name, conn)
 
-    # Close the connection
-        pass
+        # Close the connection
         conn.close()
     else:
         print("Failed to create database connection.")
 
-
 if __name__ == '__main__':
     main()