Improve database initialization with better error handling and startup debugging

app.py

@@ -20,17 +20,30 @@ def main():
     os.makedirs("app/static/images/dishes", exist_ok=True)
     os.makedirs("templates", exist_ok=True)
     
-    # Ensure database file is in the correct location for Docker
+    # Debug: Check current directory and files
+    import os
+    print(f"🔍 Current working directory: {os.getcwd()}")
+    print(f"🔍 Files in current directory: {os.listdir('.')}")
+    
+    # Check for database file
+    db_files = [f for f in os.listdir('.') if f.endswith('.db')]
+    print(f"🔍 Database files found: {db_files}")
+    
+    # Ensure database file is accessible
     source_db = "Tabble.db"
-    target_db = "Tabble.db"  # Same location in Docker
     
     try:
         if os.path.exists(source_db):
-            print(f"✅ Database found at {source_db}")
+            size = os.path.getsize(source_db)
+            print(f"✅ Database found: {source_db} ({size} bytes)")
+            # Check if file is readable
+            with open(source_db, 'rb') as f:
+                f.read(16)  # Try to read first 16 bytes
+            print(f"✅ Database file is readable")
         else:
             print(f"⚠️  Database not found at {source_db}, will create new one")
     except Exception as e:
-        print(f"⚠️  Database setup warning: {e}")
+        print(f"⚠️  Database file check error: {e}")
     
     # Set environment variables for Hugging Face Spaces
     os.environ.setdefault("HUGGINGFACE_SPACES", "1")

app/database.py

@@ -409,28 +409,41 @@ def get_current_database():
 # Create tables
 def create_tables():
     try:
-        # Ensure database directory exists (current working directory)
-        import os
-        # We're already in /app in Docker, so just ensure file can be created
-        
         # Create all tables (only creates tables that don't exist)
         Base.metadata.create_all(bind=engine)
         print("Database tables created/verified successfully")
     except Exception as e:
         print(f"Error creating database tables: {e}")
-        # Try to create a new database file if the original doesn't exist
+        print("Attempting to create new database with proper initialization...")
+        
         try:
-            print("Attempting to create new database...")
+            # Create a new SQLite database file in the current directory
             import sqlite3
-            # Create the database file if it doesn't exist (in current directory)
-            with sqlite3.connect("Tabble.db") as conn:
-                conn.execute("SELECT 1")
+            import os
+            
+            # Ensure we can write to the current directory
+            db_path = "Tabble.db"
+            
+            # Create the database file with basic structure
+            conn = sqlite3.connect(db_path)
+            conn.execute("PRAGMA journal_mode=WAL;")
+            conn.execute("PRAGMA synchronous=NORMAL;")
+            conn.close()
+            
+            print(f"Created new database file: {db_path}")
+            
+            # Recreate engine with the new database
+            global engine
+            engine.dispose()
+            engine = create_engine(f"sqlite:///{db_path}", connect_args={"check_same_thread": False})
+            
             # Try creating tables again
             Base.metadata.create_all(bind=engine)
-            print("Database created successfully")
+            print("Database tables created successfully in new database")
+            
         except Exception as creation_error:
-            print(f"Failed to create database: {creation_error}")
-            raise
+            print(f"Failed to create new database: {creation_error}")
+            print("Starting without database - some features may not work")
 
 
 # Get database session (legacy)

app/main.py

@@ -43,8 +43,14 @@ app.include_router(table.router)
 app.include_router(analytics.router)
 app.include_router(settings.router)
 
-# Create database tables
-create_tables()
+# Database initialization on startup
+@app.on_event("startup")
+async def startup_event():
+    """Initialize database tables on application startup"""
+    try:
+        create_tables()
+    except Exception as e:
+        print(f"Warning: Database initialization failed: {e}")
 
 # Check if we have the React build folder
 react_build_dir = "frontend/build"