Upload 10 files

main.py

@@ -18,9 +18,19 @@ app = FastAPI(
 # Configure CORS
 origins = [
     os.getenv("FRONTEND_URL", "http://localhost:3000"),
-    "https://app.seamo.earth",
+    "https://seamo.earth",
+    "https://seamoo.netlify.app",
+    "https://seamo-ai-ai-server.hf.space",
+    "https://seamo-ai-auth-server.hf.space",
+    "https://seamo-ai-scraper-server.hf.space",
+    "http://localhost:3000",  # For local development
+    "http://localhost:8000",  # Local AI server
+    "http://localhost:8001",  # Local Auth server
+    "http://localhost:8002",  # Local Scraper server
 ]
 
+print(f"CORS origins configured: {origins}")
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=origins,
@@ -32,56 +42,77 @@ app.add_middleware(
 # Database connection
 @app.on_event("startup")
 async def startup_db_client():
-    app.db_conn = libsql.connect(
-        "auth_server.db",
-        sync_url=os.getenv("TURSO_DATABASE_URL"),
-        auth_token=os.getenv("TURSO_AUTH_TOKEN")
-    )
-    
-    # Create tables if they don't exist
-    app.db_conn.execute("""
-    CREATE TABLE IF NOT EXISTS users (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        email TEXT NOT NULL UNIQUE CHECK(LENGTH(email) <= 320),
-        hashed_password TEXT NOT NULL CHECK(LENGTH(hashed_password) = 97),
-        created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-        last_login DATETIME,
-        is_admin BOOLEAN DEFAULT FALSE
-    )
-    """)
-    
-    app.db_conn.execute("""
-    CREATE TABLE IF NOT EXISTS projects (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        owner_id INTEGER NOT NULL,
-        title TEXT NOT NULL CHECK(LENGTH(title) BETWEEN 3 AND 120),
-        description TEXT CHECK(LENGTH(description) <= 2000),
-        geojson TEXT,
-        storage_bucket TEXT DEFAULT 'default',
-        created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-        updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-        FOREIGN KEY (owner_id) REFERENCES users(id)
-    )
-    """)
-    
-    app.db_conn.execute("""
-    CREATE TABLE IF NOT EXISTS journals (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        project_id INTEGER NOT NULL,
-        title TEXT NOT NULL CHECK(LENGTH(title) BETWEEN 3 AND 120),
-        content TEXT CHECK(LENGTH(content) <= 50000),
-        created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-        updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-        FOREIGN KEY (project_id) REFERENCES projects(id)
-    )
-    """)
-    
-    app.db_conn.commit()
-    app.db_conn.sync()
+    # Get environment variables
+    db_url = os.getenv("TURSO_DATABASE_URL")
+    auth_token = os.getenv("TURSO_AUTH_TOKEN")
+
+    if not db_url or not auth_token:
+        error_msg = "Missing Turso credentials. TURSO_DATABASE_URL and TURSO_AUTH_TOKEN must be set."
+        print(error_msg)
+        raise Exception(error_msg)
+
+    try:
+        # Use the direct URL connection method (recommended for Hugging Face)
+        print(f"Connecting to Turso database at: {db_url}")
+        app.db_conn = libsql.connect(db_url, auth_token=auth_token.strip())
+
+        # Test connection
+        result = app.db_conn.execute("SELECT 1").fetchone()
+        print(f"Connection test successful: {result}")
+
+        # Create tables if they don't exist
+        app.db_conn.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            email TEXT NOT NULL UNIQUE CHECK(LENGTH(email) <= 320),
+            hashed_password TEXT NOT NULL CHECK(LENGTH(hashed_password) = 97),
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            last_login DATETIME,
+            is_admin BOOLEAN DEFAULT FALSE
+        )
+        """)
+
+        app.db_conn.execute("""
+        CREATE TABLE IF NOT EXISTS projects (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            owner_id INTEGER NOT NULL,
+            title TEXT NOT NULL CHECK(LENGTH(title) BETWEEN 3 AND 120),
+            description TEXT CHECK(LENGTH(description) <= 2000),
+            geojson TEXT,
+            storage_bucket TEXT DEFAULT 'default',
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (owner_id) REFERENCES users(id)
+        )
+        """)
+
+        app.db_conn.execute("""
+        CREATE TABLE IF NOT EXISTS journals (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            project_id INTEGER NOT NULL,
+            title TEXT NOT NULL CHECK(LENGTH(title) BETWEEN 3 AND 120),
+            content TEXT CHECK(LENGTH(content) <= 50000),
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (project_id) REFERENCES projects(id)
+        )
+        """)
+
+        app.db_conn.commit()
+        print("Database tables created successfully")
+
+    except Exception as e:
+        error_msg = f"Failed to connect to Turso database: {str(e)}"
+        print(error_msg)
+        raise Exception(error_msg)
 
 @app.on_event("shutdown")
 async def shutdown_db_client():
-    app.db_conn.close()
+    try:
+        app.db_conn.close()
+        print("Database connection closed successfully")
+    except Exception as e:
+        print(f"Error closing database connection: {str(e)}")
 
 # Root endpoint
 @app.get("/")