Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import os
 import shutil
-import json
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 import urllib.parse
@@ -25,6 +24,16 @@ theme = gr.themes.Default(
     primary_hue="blue",
     secondary_hue="violet",
     neutral_hue="slate",
+).set(
+    button_secondary_background_fill='*secondary_200',
+    button_secondary_background_fill_hover='*secondary_300',
+    button_secondary_background_fill_dark='*secondary_600',
+    button_secondary_background_fill_hover_dark='*secondary_500',
+    button_secondary_border_color='*secondary_200',
+    button_secondary_border_color_hover='*secondary_300',
+    button_secondary_border_color_dark='*secondary_600',
+    button_secondary_border_color_hover_dark='*secondary_500',
+    button_secondary_text_color='*secondary_700'
 )
 
 def upload(file, secret):
@@ -57,11 +66,13 @@ def get_folder_contents_dict(root_dir):
 def generate_markdown(json_data):
     urls = json_data.get("urls", [])
     markdown_lines = []
+    count = 1
     for url in urls:
         encoded_url = urllib.parse.quote(url, safe=':/')
         filename = url.split('/')[-1]
-        encoded_filename = urllib.parse.quote(filename)
-        markdown_lines.append(f'- <a style="text-decoration:none; font-size:18px" href={encoded_url} target="_blank">{encoded_filename}</a>')
+        encoded_filename = urllib.parse.quote(filename).replace("%20", " ")
+        markdown_lines.append(f'> {count}. <a style="text-decoration:none;" href={encoded_url} target="_blank">{encoded_filename}</a>')
+        count += 1
     return "\n".join(markdown_lines)
 
 def get_uploads(secret):
@@ -80,7 +91,7 @@ def get_notes(secret):
     if secret == os.environ.get('SECRET_KEY'):
         result = references_collection.find_one({"_id": ObjectId('6655fe4ca6913aa201819e72')}, {"_id": 0})
         if result:
-            gr.Info("Recieved uploaded files")
+            gr.Info("Recieved shared notes")
             return result['notes']
         else:
             gr.Info("No result found")
@@ -102,6 +113,44 @@ def save_notes(note, secret):
         gr.Warning("Invalid Secret Key")
         return note
 
+def delete_path(path):
+    try:
+        if os.path.isfile(path):
+            os.remove(path)
+            filename = os.path.basename(path)
+            gr.Info(f"The file {filename} has been deleted successfully.")
+        elif os.path.isdir(path):
+            shutil.rmtree(path)
+            foldername = os.path.basename(path)
+            gr.Info(f"The directory {foldername} has been deleted successfully.")
+        else:
+            print(f"The path {path} does not exist.")
+            gr.Warning("Invalid action.")
+    except Exception as e:
+        print(f"An error occurred while deleting {path}: {e}")
+        gr.Warning("Invalid action.")
+
+def delete_files(path, secret):
+    if secret == os.environ.get('SECRET_KEY'):
+        for paths in path:
+            delete_path(paths)
+    else:
+        gr.Warning("Invalid Secret Key")
+
+    return gr.update(root_dir="/tmp/gradio")
+
+def toogle_visibility(secret):
+    if secret == os.environ.get('SECRET_KEY'):
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
+    else:
+        raise gr.Error("Invalid Secret Key")
+
+def toogle_vis():
+    return gr.update(visible=True)
+
+def toogle_vis2():
+    return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+
 js = '''
 function test() {
     document.title = "File Sharing System";
@@ -137,6 +186,10 @@ function test() {
 '''
 
 css = '''
+body::-webkit-scrollbar {
+    display: none;
+    scroll-behavior: smooth;
+}
 .hero-container {
     display: flex;
     flex-direction: row;
@@ -170,22 +223,32 @@ with gr.Blocks(theme=theme, js=js, css=css) as demo:
     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 = gr.File(label="Upload", file_count="multiple")
             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>''')
             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)
+            initial_delete_button = gr.Button("Delete Uploads?", variant="stop", visible=True)
+            cancel_button = gr.Button("Cancel", variant="secondary", visible=False)
+        with gr.Column(visible=False) as temp:
+            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">Navigator</h1>''')
+            file_exe = gr.FileExplorer(root_dir="/tmp/gradio")
+            final_delete_button = gr.Button("Delete", variant="stop", visible=False)
+            final_delete_button.click(fn=delete_files, inputs=[file_exe, secret_key], outputs=file_exe)
+            file_exe.change(fn=toogle_vis, outputs=final_delete_button)
+        initial_delete_button.click(fn=toogle_visibility, inputs=secret_key, outputs=[initial_delete_button, cancel_button, temp])
+        cancel_button.click(fn=toogle_vis2, outputs=[temp, initial_delete_button, cancel_button])
+        
             
     notes = gr.TextArea(label="📝 Share Notes", placeholder="Enter your notes...")
     with gr.Row():    
-        get_notes_button = gr.Button("Get Notes", variant="primary")
+        get_notes_button = gr.Button("Get Notes", variant="secondary")
         get_notes_button.click(fn=get_notes, inputs=secret_key, outputs=notes)
         save_notes_button = gr.Button("Save", variant="primary")
         save_notes_button.click(fn=save_notes, inputs=[notes,secret_key], outputs=notes)
 
     gr.Markdown('''> To know more read the docs at: [Documentation](https://huggingface.co/spaces/abhicodes/file-sharing-system/blob/main/README.md)''')
 
-
 demo.launch(debug=True)