Spaces:
Runtime error
Runtime error
File size: 1,519 Bytes
14310b1 5fb1a7a 14310b1 5fb1a7a 14310b1 5fb1a7a 14310b1 5fb1a7a 14310b1 5fb1a7a 14310b1 5fb1a7a 14310b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import os
import gradio as gr
from werkzeug.utils import secure_filename
UPLOAD_FOLDER = './'
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'doc', 'docx', 'png', 'jpg', 'jpeg', 'gif'}
app = gr.Interface(
fn=None,
inputs=None,
outputs=None,
title='Workspace Uploader',
description='Upload documents into a workspace',
theme='default',
layout='wide',
allow_flagging=False,
analytics_enabled=False,
server_name=None,
)
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
def upload_file(request):
if request.method == 'POST':
# check if the post request has the file part
if 'file' not in request.files:
return 'No file part'
file = request.files['file']
# if user does not select file, browser also
# submit an empty part without filename
if file.filename == '':
return 'No selected file'
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
file.save(os.path.join(UPLOAD_FOLDER, filename))
return 'File uploaded successfully'
else:
return 'File not allowed'
@app.interface(
"textbox",
"file",
live=True
)
def upload_to_workspace(text, file):
if file is not None:
upload_file(file)
return 'File uploaded successfully'
else:
return 'No file selected'
if __name__ == '__main__':
app.launch()
|