Spaces:

albumup
/

file

Sleeping

App Files Files Community

albumup commited on Jan 17, 2025

Commit

b2de22f

verified ·

1 Parent(s): 503d684

Update app.py

Browse files

Files changed (1) hide show

app.py +136 -8

app.py CHANGED Viewed

@@ -21,15 +21,50 @@ HTML_CONTENT = """
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>File Upload</title>
     <style>
-        .container { max-width: 800px; margin: 0 auto; padding: 20px; }
-        .upload-form { margin-bottom: 20px; }
-        .preview { margin-top: 20px; }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>File Upload</h1>
         <!-- Single File Upload -->
         <div class="upload-form">
             <h2>Single File Upload</h2>
@@ -59,7 +94,7 @@ ALBUM_VIEW_HTML = """
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Album View</title>
     <style>
         .container {{
             max-width: 800px;
@@ -75,17 +110,35 @@ ALBUM_VIEW_HTML = """
             border: 1px solid #ddd;
             padding: 10px;
             text-align: center;
         }}
         .download-all {{
             margin: 20px 0;
         }}
     </style>
 </head>
 <body>
     <div class="container">
         <h1>{album_name}</h1>
         <div class="download-all">
-            <a href="/album/{album_id}/download" class="button">Download All Files</a>
         </div>
         <div class="file-grid">
             {file_list}
@@ -95,10 +148,85 @@ ALBUM_VIEW_HTML = """
 </html>
 """
 @app.get("/", response_class=HTMLResponse)
 async def index():
     return HTML_CONTENT
 @app.post("/upload")
 async def handle_upload(request: Request, file: UploadFile = File(...)):
     if not file:
@@ -168,8 +296,8 @@ async def view_album(album_id: str):
         file_list_html += f"""
         <div class="file-item">
             <p>{file['filename']}</p>
-            <a href="/upload/{file['path']}" target="_blank">View</a>
-            <a href="/upload/{file['path']}?download=true">Download</a>
         </div>
         """

     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>File Upload</title>
     <style>
+        .container {{
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }}
+        .upload-form {{
+            margin-bottom: 20px;
+            padding: 20px;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+        }}
+        .preview {{
+            margin-top: 20px;
+        }}
+        .search-form {{
+            margin: 20px 0;
+            padding: 20px;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+        }}
+        .album-list {{
+            margin-top: 20px;
+        }}
+        .album-item {{
+            padding: 10px;
+            border: 1px solid #ddd;
+            margin-bottom: 10px;
+            border-radius: 5px;
+        }}
     </style>
 </head>
 <body>
     <div class="container">
+        <h1>File Upload & Album Management</h1>
+        <!-- Search Albums -->
+        <div class="search-form">
+            <h2>Search Albums</h2>
+            <form action="/search" method="get">
+                <input type="text" name="query" placeholder="Search by album name..." required>
+                <button type="submit">Search</button>
+            </form>
+        </div>
         <!-- Single File Upload -->
         <div class="upload-form">
             <h2>Single File Upload</h2>
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Album View - {album_name}</title>
     <style>
         .container {{
             max-width: 800px;
             border: 1px solid #ddd;
             padding: 10px;
             text-align: center;
+            border-radius: 5px;
         }}
         .download-all {{
             margin: 20px 0;
         }}
+        .back-link {{
+            margin-bottom: 20px;
+            display: block;
+        }}
+        .button {{
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #007bff;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin: 5px;
+        }}
+        .button:hover {{
+            background-color: #0056b3;
+        }}
     </style>
 </head>
 <body>
     <div class="container">
+        <a href="/" class="back-link button">← Back to Home</a>
         <h1>{album_name}</h1>
         <div class="download-all">
+            <a href="/album/{album_id}/download" class="button" target="_blank">Download All Files</a>
         </div>
         <div class="file-grid">
             {file_list}
 </html>
 """
+SEARCH_RESULTS_HTML = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Search Results</title>
+    <style>
+        .container {{
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }}
+        .album-item {{
+            border: 1px solid #ddd;
+            padding: 15px;
+            margin-bottom: 15px;
+            border-radius: 5px;
+        }}
+        .back-link {{
+            margin-bottom: 20px;
+            display: block;
+        }}
+        .button {{
+            display: inline-block;
+            padding: 8px 16px;
+            background-color: #007bff;
+            color: white;
+            text-decoration: none;
+            border-radius: 4px;
+            margin: 5px;
+        }}
+        .button:hover {{
+            background-color: #0056b3;
+        }}
+    </style>
+</head>
+<body>
+    <div class="container">
+        <a href="/" class="back-link button">← Back to Home</a>
+        <h1>Search Results</h1>
+        <p>Found {result_count} albums for "{query}"</p>
+        <div class="search-results">
+            {results}
+        </div>
+    </div>
+</body>
+</html>
+"""
 @app.get("/", response_class=HTMLResponse)
 async def index():
     return HTML_CONTENT
+@app.get("/search", response_class=HTMLResponse)
+async def search_albums(query: str = ""):
+    query = query.lower()
+    matching_albums = {
+        album_id: album for album_id, album in albums.items()
+        if query in album['name'].lower()
+    }
+    results_html = ""
+    for album_id, album in matching_albums.items():
+        results_html += f"""
+        <div class="album-item">
+            <h3>{album['name']}</h3>
+            <p>Created: {album['created_at']}</p>
+            <p>Files: {len(album['files'])}</p>
+            <a href="/album/{album_id}" class="button">View Album</a>
+        </div>
+        """
+    return SEARCH_RESULTS_HTML.format(
+        query=query,
+        result_count=len(matching_albums),
+        results=results_html
+    )
 @app.post("/upload")
 async def handle_upload(request: Request, file: UploadFile = File(...)):
     if not file:
         file_list_html += f"""
         <div class="file-item">
             <p>{file['filename']}</p>
+            <a href="/upload/{file['path']}" class="button">View</a>
+            <a href="/upload/{file['path']}?download=true" class="button" target="_blank">Download</a>
         </div>
         """