Create app.py

app.py

@@ -0,0 +1,43 @@
+import gradio as gr
+import os
+import sqlite3
+import uuid
+
+UPLOADS_FOLDER = 'uploads'
+
+if not os.path.exists(UPLOADS_FOLDER):
+    os.mkdir(UPLOADS_FOLDER)
+
+conn = sqlite3.connect('documents.db')
+c = conn.cursor()
+c.execute('''CREATE TABLE IF NOT EXISTS documents
+             (id INTEGER PRIMARY KEY AUTOINCREMENT,
+              name TEXT NOT NULL,
+              user_id TEXT NOT NULL,
+              file_path TEXT NOT NULL)''')
+conn.commit()
+
+def save_file(upload_file, user_id):
+    user_folder = os.path.join(UPLOADS_FOLDER, user_id)
+    if not os.path.exists(user_folder):
+        os.mkdir(user_folder)
+    file_path = os.path.join(user_folder, upload_file.name)
+    with open(file_path, 'wb') as f:
+        f.write(upload_file.read())
+    return file_path
+
+def save_to_database(file_path, user_id):
+    name = os.path.basename(file_path)
+    c.execute("INSERT INTO documents (name, user_id, file_path) VALUES (?, ?, ?)", (name, user_id, file_path))
+    conn.commit()
+    return "File saved successfully!"
+
+def upload_and_save_file(upload_file, user_id):
+    file_path = save_file(upload_file, user_id)
+    save_to_database(file_path, user_id)
+    return "File uploaded and saved to database successfully!"
+
+inputs = [gr.inputs.File(label="Upload a file"), gr.inputs.Hidden(label="User ID")]
+outputs = gr.outputs.Textbox(label="Status")
+
+gr.Interface(upload_and_save_file, inputs, outputs, title="Document Uploader", allow_screenshot=False).launch()