Update main.py

main.py

@@ -22,14 +22,15 @@ app = Flask(__name__)
 CORS(app)
 
 # --- App Configuration ---
-# --- FIX 1: Switched to a persistent file-based SQLite database ---
-# This ensures data survives between requests on Hugging Face Spaces.
-DB_FOLDER = 'data'
+# --- FIX: Use an absolute path for the database to ensure correct file location on server environments ---
+basedir = os.path.abspath(os.path.dirname(__file__))
+DB_FOLDER = os.path.join(basedir, 'data')
 DB_PATH = os.path.join(DB_FOLDER, 'products.db')
-os.makedirs(DB_FOLDER, exist_ok=True)
+os.makedirs(DB_FOLDER, exist_ok=True) # Ensure the data directory exists
+
 app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{DB_PATH}'
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
-app.config['UPLOAD_FOLDER'] = 'uploads'
+app.config['UPLOAD_FOLDER'] = os.path.join(basedir, 'uploads') # Also make upload folder absolute
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 
 # --- File Upload Configuration ---
@@ -124,8 +125,6 @@ def process_uploaded_file(filepath, filename):
 
     try:
         file_ext = filename.rsplit('.', 1)[1].lower()
-        # --- FIX 2: Robustly parse the second column (index 1) for names ---
-        # The user's uploaded `list.csv` clearly has the product name in the second column.
         if file_ext == 'csv':
             df = pd.read_csv(filepath, header=None, usecols=[1], names=['product_name'])
         elif file_ext in ['xls', 'xlsx']:
@@ -166,7 +165,6 @@ def process_uploaded_file(filepath, filename):
             "primary_category": best_hs_desc or "N/A", "status": ""
         }
         try:
-            # Each operation needs its own app context to interact with the database
             with app.app_context():
                 existing_product = Product.query.filter_by(name=validated_name).first()
                 if existing_product: