Spaces:
Runtime error
Runtime error
Commit
·
27abc69
1
Parent(s):
d89df63
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,43 +1,21 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import os
|
| 3 |
-
import sqlite3
|
| 4 |
-
import uuid
|
| 5 |
|
| 6 |
-
|
| 7 |
|
| 8 |
-
if not os.path.exists(
|
| 9 |
-
os.mkdir(
|
| 10 |
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
c.execute('''CREATE TABLE IF NOT EXISTS documents
|
| 14 |
-
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
| 15 |
-
name TEXT NOT NULL,
|
| 16 |
-
user_id TEXT NOT NULL,
|
| 17 |
-
file_path TEXT NOT NULL)''')
|
| 18 |
-
conn.commit()
|
| 19 |
-
|
| 20 |
-
def save_file(upload_file, user_id):
|
| 21 |
-
user_folder = os.path.join(UPLOADS_FOLDER, user_id)
|
| 22 |
-
if not os.path.exists(user_folder):
|
| 23 |
-
os.mkdir(user_folder)
|
| 24 |
-
file_path = os.path.join(user_folder, upload_file.name)
|
| 25 |
with open(file_path, 'wb') as f:
|
| 26 |
f.write(upload_file.read())
|
| 27 |
-
return file_path
|
| 28 |
-
|
| 29 |
-
def save_to_database(file_path, user_id):
|
| 30 |
-
name = os.path.basename(file_path)
|
| 31 |
-
c.execute("INSERT INTO documents (name, user_id, file_path) VALUES (?, ?, ?)", (name, user_id, file_path))
|
| 32 |
-
conn.commit()
|
| 33 |
return "File saved successfully!"
|
| 34 |
|
| 35 |
-
def upload_and_save_file(upload_file
|
| 36 |
-
|
| 37 |
-
save_to_database(file_path, user_id)
|
| 38 |
-
return "File uploaded and saved to database successfully!"
|
| 39 |
|
| 40 |
-
inputs =
|
| 41 |
outputs = gr.outputs.Textbox(label="Status")
|
| 42 |
|
| 43 |
-
gr.Interface(upload_and_save_file, inputs, outputs, title="Document Uploader"
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import os
|
|
|
|
|
|
|
| 3 |
|
| 4 |
+
UPLOAD_FOLDER = 'uploads'
|
| 5 |
|
| 6 |
+
if not os.path.exists(UPLOAD_FOLDER):
|
| 7 |
+
os.mkdir(UPLOAD_FOLDER)
|
| 8 |
|
| 9 |
+
def save_file(upload_file):
|
| 10 |
+
file_path = os.path.join(UPLOAD_FOLDER, upload_file.name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
with open(file_path, 'wb') as f:
|
| 12 |
f.write(upload_file.read())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
return "File saved successfully!"
|
| 14 |
|
| 15 |
+
def upload_and_save_file(upload_file):
|
| 16 |
+
return save_file(upload_file)
|
|
|
|
|
|
|
| 17 |
|
| 18 |
+
inputs = gr.inputs.File(label="Upload a file")
|
| 19 |
outputs = gr.outputs.Textbox(label="Status")
|
| 20 |
|
| 21 |
+
gr.Interface(upload_and_save_file, inputs, outputs, title="Document Uploader").launch()
|