Update app.py

app.py

@@ -76,6 +76,32 @@ def get_uploads(secret):
     else:
         return '''<p style="color:red;font-size:20px;text-align:center;font-weight:bold;">Invalid Secret Key</p>'''
 
+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")
+            return result['notes']
+        else:
+            gr.Info("No result found")
+            return '''No result found'''
+    else:
+        gr.Warning("Invalid Secret Key")
+        return '''Invalid Secret Key'''
+
+def save_notes(note, secret):
+    if secret == os.environ.get('SECRET_KEY'):
+        result = references_collection.update_one({"_id": ObjectId('6655fe4ca6913aa201819e72')}, {"$set": {"notes": note}})
+        if result:
+            gr.Info("Notes saved successfully.")
+            return note
+        else:
+            gr.Info("No result found")
+            return '''No result found'''
+    else:
+        gr.Warning("Invalid Secret Key")
+        return note
+
 js = '''
 function test() {
     document.title = "File Sharing System";
@@ -151,6 +177,13 @@ with gr.Blocks(theme=theme, js=js, css=css) as demo:
             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)
+            
+    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.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)''')