Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import shutil
 import json
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
@@ -26,20 +27,22 @@ theme = gr.themes.Default(
     neutral_hue="slate",
 )
 
-def upload(file):
-    root_directory = '/tmp/gradio'
-    folder_contents_dict = get_folder_contents_dict(root_directory)
-    print(folder_contents_dict)
-
-    base_url = "https://abhicodes-file-sharing-system.hf.space/file=/tmp/gradio"
-
-    urls = [f"{base_url}/{folder}/{file}" for folder, files in folder_contents_dict.items() for file in files]
-
-    print(json.dumps(urls, indent=4))
-
-    references_collection.update_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"$set": {"urls": urls}})
+def upload(file, secret):
+    if secret == os.environ.get('SECRET_KEY'):
+        root_directory = '/tmp/gradio'
+        folder_contents_dict = get_folder_contents_dict(root_directory)
+        base_url = "https://abhicodes-file-sharing-system.hf.space/file=/tmp/gradio"
+        urls = [f"{base_url}/{folder}/{file}" for folder, files in folder_contents_dict.items() for file in files]
+        references_collection.update_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"$set": {"urls": urls}})
 
-    gr.Info("Uploaded Successfully")
+        gr.Info("Uploaded Successfully")
+    
+    else:
+        for ele in file:
+            dir_to_remove = os.path.dirname(ele)
+            shutil.rmtree(dir_to_remove)
+        
+        gr.Warning("Unauthorized to upload")
 
 def get_folder_contents_dict(root_dir):
     folder_contents = {}
@@ -55,7 +58,6 @@ def generate_markdown(json_data):
     urls = json_data.get("urls", [])
     markdown_lines = []
     for url in urls:
-        # Encode URL components
         encoded_url = urllib.parse.quote(url, safe=':/')
         filename = url.split('/')[-1]
         encoded_filename = urllib.parse.quote(filename)
@@ -67,39 +69,85 @@ def get_uploads(secret):
         result = references_collection.find_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"_id": 0})
         if result:
             markdown_output = generate_markdown(result)
+            gr.Info("Recieved uploaded files")
             return markdown_output
         else:
-            return {"error": "No result found"}
+            return '''<p style="color:red;font-size:20px;text-align:center;font-weight:bold;">No result found</p>'''
     else:
-        return {"error": "Unauthorized"}
+        return '''<p style="color:red;font-size:20px;text-align:center;font-weight:bold;">Invalid Secret Key</p>'''
 
 js = '''
 function test() {
-    console.log("Test");
     document.title = "File Sharing System";
     var link = document.createElement('link');
     link.type = 'image/x-icon';
     link.rel = 'shortcut icon';
     link.href = 'https://cdn3d.iconscout.com/3d/premium/thumb/cloud-storage-5402862-4521475.png';
     document.getElementsByTagName('head')[0].appendChild(link);
+
+    function updateHeroContainerStyle() {
+        if (window.innerWidth <= 600) {
+            const ele = document.querySelector(".hero-container");
+            if (ele) {
+                ele.style.flexDirection = 'column';
+                ele.style.padding = '5px';
+                ele.style.margin = '0px';
+                ele.style.gap = '0px';
+            }
+        }
+        else {
+            const ele = document.querySelector(".hero-container");
+            if (ele) {
+                ele.style.flexDirection = 'row';
+                ele.style.padding = '5px';
+                ele.style.margin = '0px 20px 0px 20px';
+                ele.style.gap = '50px';
+            }
+        }
+    }
+    updateHeroContainerStyle();
+    window.addEventListener('resize', updateHeroContainerStyle);
 }
 '''
 
-with gr.Blocks(theme=theme, js=js) as demo:
+css = '''
+.hero-container {
+    display: flex;
+    flex-direction: row;
+    gap:50px;
+    justify-content: center;
+    align-items: center;
+    padding: 5px;
+    border-radius: 10px;
+    max-width: 100%;
+    margin: 20px;
+    margin-top: 0px;
+    margin-bottom: 0px;
+}
+.logo {
+    width: 80px;
+}
+.description {
+    text-align: justify;
+}
+'''
+
+with gr.Blocks(theme=theme, js=js, css=css) as demo:
     gr.Markdown('''<h1 style="text-align:center;">File Storing and Sharing System</h1>''')
     with gr.Column():
-        gr.Markdown('''<div style="display:flex;justify-content:center;align-items:center"><img src="https://cdn3d.iconscout.com/3d/premium/thumb/cloud-storage-5402862-4521475.png" alt="logo" width="100"/>
-                        <div style="width:100px;"></div>
-                        This project is a file storing and sharing system built using Gradio for the front-end interface and MongoDB for the back-end database. The system allows users to upload files, which are then stored and managed within a specified directory. URLs for accessing these files are generated and stored in a MongoDB collection, making it easy to share and access the uploaded files.''')
-    
+        gr.Markdown(''' <div class="hero-container">
+                            <img src="https://cdn3d.iconscout.com/3d/premium/thumb/cloud-storage-5402862-4521475.png" alt="logo" class="logo">
+                            <p class="description">This project is a file storing and sharing system built using Gradio for the front-end interface and MongoDB for the back-end database. The system allows users to upload files, which are then stored and managed within a specified directory. URLs for accessing these files are generated and stored in a MongoDB collection, making it easy to share and access the uploaded files.</p>
+                        </div> ''')    
+        
+    secret_key = gr.Text(label="🔐 Secret Key", placeholder="Enter your secret key here...", type="password", autofocus=True)
     with gr.Row():
         with gr.Column():
             gr.Markdown('''<h1 style="text-align:center;display:flex;justify-content:center;align-items:center"><img src="https://i.ibb.co/FB6tMdT/upload.png" alt="upload" width="50">Uploader</h1>''')
             input = gr.File(label="Input", file_count="multiple")
-            input.upload(fn=upload, inputs=input)
+            input.upload(fn=upload, inputs=[input, secret_key])
         with gr.Column():
             gr.Markdown('''<h1 style="text-align:center;display:flex;justify-content:center;align-items:center"><img src="https://i.ibb.co/K9Bq3j7/download.png" alt="download" width="50">Downloader</h1>''')
-            secret_key = gr.Text(placeholder="Enter your Secret Key", type="password", autofocus=True)
             uploads = gr.Markdown(label="Uploads")
             get_upload_button = gr.Button("Get Uploads", variant='primary')
             get_upload_button.click(fn=get_uploads, inputs=secret_key, outputs=uploads)