Update app.py

app.py

@@ -1,93 +1,93 @@
-import gradio as gr
-import os
-import json
-from pymongo.mongo_client import MongoClient
-from pymongo.server_api import ServerApi
-import urllib.parse
-from bson.objectid import ObjectId
-
-username = urllib.parse.quote_plus(os.getenv('MONGO_USERNAME'))
-password = urllib.parse.quote_plus(os.getenv('MONGO_PASSWORD'))
-restUri = os.getenv('REST_URI')
-uri = f'mongodb+srv://{username}:{password}{restUri}'
-client = MongoClient(uri, server_api=ServerApi('1'))
-db = client['file_storage']
-references_collection = db['references']
-
-try:
-    client.admin.command('ping')
-    print("Pinged your deployment. You successfully connected to MongoDB!")
-except Exception as e:
-    print(e)
-
-theme = gr.themes.Default(
-    primary_hue="blue",
-    secondary_hue="violet",
-    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-uploader-component.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}})
-
-    gr.Info("Uploaded Successfully")
-
-def get_folder_contents_dict(root_dir):
-    folder_contents = {}
-    for item in os.listdir(root_dir):
-        item_path = os.path.join(root_dir, item)
-        if os.path.isdir(item_path):
-            contents = os.listdir(item_path)
-            folder_contents[item] = contents
-    
-    return folder_contents
-
-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)
-        markdown_lines.append(f'- <a style="text-decoration:none; font-size:18px" href={encoded_url} target="_blank">{encoded_filename}</a>')
-    return "\n".join(markdown_lines)
-
-def get_uploads(secret):
-    if secret == os.environ.get('SECRET_KEY'):
-        result = references_collection.find_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"_id": 0})
-        if result:
-            markdown_output = generate_markdown(result)
-            return markdown_output
-        else:
-            return {"error": "No result found"}
-    else:
-        return {"error": "Unauthorized"}
-
-with gr.Blocks(theme=theme) as demo:
-    gr.Markdown('''<h1 style="text-align:center;">File Storing and Sharing System</h1>''')
-    gr.Markdown('''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.
-                > To know more read the docs at: [Documentation](https://huggingface.co/spaces/abhicodes/file-sharing-system/blob/main/README.md)
-                ''')
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown('''<h1 style="text-align:center;">Uploader</h1>''')
-            input = gr.File(label="Input", file_count="multiple")
-            input.upload(fn=upload, inputs=input)
-        with gr.Column():
-            gr.Markdown('''<h1 style="text-align:center;">Downloader</h1>''')
-            secret_key = gr.Text(placeholder="Enter your Secret Key")
-            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)
-
+import gradio as gr
+import os
+import json
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+import urllib.parse
+from bson.objectid import ObjectId
+
+username = urllib.parse.quote_plus(os.getenv('MONGO_USERNAME'))
+password = urllib.parse.quote_plus(os.getenv('MONGO_PASSWORD'))
+restUri = os.getenv('REST_URI')
+uri = f'mongodb+srv://{username}:{password}{restUri}'
+client = MongoClient(uri, server_api=ServerApi('1'))
+db = client['file_storage']
+references_collection = db['references']
+
+try:
+    client.admin.command('ping')
+    print("Pinged your deployment. You successfully connected to MongoDB!")
+except Exception as e:
+    print(e)
+
+theme = gr.themes.Default(
+    primary_hue="blue",
+    secondary_hue="violet",
+    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}})
+
+    gr.Info("Uploaded Successfully")
+
+def get_folder_contents_dict(root_dir):
+    folder_contents = {}
+    for item in os.listdir(root_dir):
+        item_path = os.path.join(root_dir, item)
+        if os.path.isdir(item_path):
+            contents = os.listdir(item_path)
+            folder_contents[item] = contents
+    
+    return folder_contents
+
+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)
+        markdown_lines.append(f'- <a style="text-decoration:none; font-size:18px" href={encoded_url} target="_blank">{encoded_filename}</a>')
+    return "\n".join(markdown_lines)
+
+def get_uploads(secret):
+    if secret == os.environ.get('SECRET_KEY'):
+        result = references_collection.find_one({"_id": ObjectId('66531a797cbaa19ba4e441c5')}, {"_id": 0})
+        if result:
+            markdown_output = generate_markdown(result)
+            return markdown_output
+        else:
+            return {"error": "No result found"}
+    else:
+        return {"error": "Unauthorized"}
+
+with gr.Blocks(theme=theme) as demo:
+    gr.Markdown('''<h1 style="text-align:center;">File Storing and Sharing System</h1>''')
+    gr.Markdown('''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.
+                > To know more read the docs at: [Documentation](https://huggingface.co/spaces/abhicodes/file-sharing-system/blob/main/README.md)
+                ''')
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown('''<h1 style="text-align:center;">Uploader</h1>''')
+            input = gr.File(label="Input", file_count="multiple")
+            input.upload(fn=upload, inputs=input)
+        with gr.Column():
+            gr.Markdown('''<h1 style="text-align:center;">Downloader</h1>''')
+            secret_key = gr.Text(placeholder="Enter your Secret Key")
+            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)
+
 demo.launch(debug=True)