Spaces:
Runtime error
Runtime error
| 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' | |
| 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() | |