Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,10 +3,10 @@ import sqlite3
 import os
 
 # ---------------- DATABASE SETUP ----------------
-DB_PATH = "telangana.db"
+DB_PATH = "/tmp/telangana.db"  # Hugging Face writable directory
 
 def create_tables():
-    
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("""
     CREATE TABLE IF NOT EXISTS districts (
@@ -32,11 +32,11 @@ def create_tables():
     conn.commit()
     conn.close()
 
-
+create_tables()
 
 # ---------------- FUNCTIONS ----------------
 def add_district(name):
-    
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("INSERT OR IGNORE INTO districts(name) VALUES(?)", (name,))
     conn.commit()
@@ -62,17 +62,14 @@ def add_product(category_id, product_name, description, image_bytes):
     conn.close()
 
 def get_all_data():
-    
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("""
-     SELECT d.name, c.category_name, 
-    p.product_name
-     FROM districts d
-     JOIN categories c ON d.id = 
-    c.district_id
-     JOIN products p ON c.id = 
-    p.category_id
-""")
+    SELECT d.name, c.category_name, p.product_name
+    FROM districts d
+    JOIN categories c ON d.id = c.district_id
+    JOIN products p ON c.id = p.category_id
+    """)
     data = cursor.fetchall()
     conn.close()
     return data
@@ -85,23 +82,19 @@ menu = st.sidebar.radio("Navigation", ["Home", "Add Data", "View by District"])
 if menu == "Home":
     st.title("🌾 Telangana Famous Products")
     st.subheader("Explore Products by District")
-if data:
-    grouped = {}
-    for district, category, product in data:
-        grouped.setdefault(district, {}).setdefault(category, []).append(product)
-
-for district, categories in grouped.items():
-    with st.expander(f"📍 {district}"):
-        for category, products in categories.items():
-            st.write(f"{category}: {', '.join(products)}")
-         
+
+    data = get_all_data()
+    if data:
+        grouped = {}
+        for district, category, product in data:
+            grouped.setdefault(district, {}).setdefault(category, []).append(product)
+
         for district, categories in grouped.items():
             with st.expander(f"📍 {district}"):
                 for category, products in categories.items():
-                    # CORRECTED LINE BELOW:
                     st.write(f"{category}: {', '.join(products)}")
-else:
-    st.info("No data available. Please add data in 'Add Data' section.")
+    else:
+        st.info("No data available. Please add data in 'Add Data' section.")
 
 # ---------------- ADD DATA PAGE ----------------
 elif menu == "Add Data":
@@ -112,15 +105,15 @@ elif menu == "Add Data":
         district_name = st.text_input("District Name")
         submitted = st.form_submit_button("Add District")
         if submitted and district_name:
-            
+            add_district(district_name)
             st.success(f"✅ District '{district_name}' added!")
 
     st.write("---")
 
     # Add Category
     with st.form("add_category_form"):
-        
-        
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
         cursor.execute("SELECT name FROM districts")
         district_list = [row[0] for row in cursor.fetchall()]
         conn.close()
@@ -206,7 +199,7 @@ elif menu == "View by District":
                             if image_data:
                                 st.image(image_data, use_column_width=True)
                         with col2:
-                            st.write(f"{product_name}") # Bolding product name for consistency
+                            st.write(f"**{product_name}**")
                             st.caption(description)
                     st.write("---")
             else: