Update app.py

app.py

@@ -8,9 +8,33 @@ import uuid
 import json
 from datetime import datetime
 import mimetypes
+import httpx
 
 app = FastAPI()
 
+DENO_API_URL = "https://dataerrr99.deno.dev"
+
+async def save_album_to_db(album_name: str, album_id: str):
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.post(DENO_API_URL, json={
+                "albumName": album_name,
+                "albumLink": album_id
+            })
+            return response.json()
+        except Exception as e:
+            print(f"Error saving to DB: {str(e)}")
+            return None
+
+async def get_albums_from_db():
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.get(DENO_API_URL)
+            return response.json()
+        except Exception as e:
+            print(f"Error fetching from DB: {str(e)}")
+            return {"data": []}
+
 # In-memory storage for albums
 albums = {}
 
@@ -302,9 +326,12 @@ async def index():
 @app.get("/search", response_class=HTMLResponse)
 async def search_albums(query: str = ""):
     query = query.lower()
+    db_albums = await get_albums_from_db()
+    
     matching_albums = {
-        album_id: album for album_id, album in albums.items()
-        if query in album['name'].lower()
+        album['link']: albums[album['link']]
+        for album in db_albums['data']
+        if album['link'] in albums and query in album['name'].lower()
     }
     
     results_html = ""
@@ -324,28 +351,6 @@ async def search_albums(query: str = ""):
         results=results_html
     )
 
-@app.post("/upload")
-async def handle_upload(request: Request, file: UploadFile = File(...)):
-    if not file:
-        return {"error": "No file uploaded"}, 400
-
-    cookies = await get_cookies()
-    if 'csrftoken' not in cookies or 'sessionid' not in cookies:
-        return {"error": "Failed to get cookies"}, 500
-
-    upload_result = await initiate_upload(cookies, file.filename, file.content_type)
-    if not upload_result or 'upload_url' not in upload_result:
-        return {"error": "Failed to initiate upload"}, 500
-
-    file_content = await file.read()
-    upload_success = await retry_upload(upload_result['upload_url'], file_content, file.content_type)
-    if not upload_success:
-        return {"error": "Upload failed"}, 500
-
-    base_url = str(request.base_url).rstrip('/')
-    redirect_url = f"{base_url}/upload/{upload_result['serving_url'].split('/pbxt/')[1]}"
-    return RedirectResponse(url=redirect_url, status_code=303)
-
 @app.post("/album/create")
 async def create_album(
     request: Request,
@@ -378,6 +383,9 @@ async def create_album(
         'files': album_files
     }
 
+    # Save to Deno KV database
+    await save_album_to_db(album_name, album_id)
+
     base_url = str(request.base_url).rstrip('/')
     return RedirectResponse(url=f"{base_url}/album/{album_id}", status_code=303)
 
@@ -506,4 +514,4 @@ async def retry_upload(upload_url, file_content, content_type, max_retries=5):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)