Update app.py

app.py

@@ -1,28 +1,55 @@
+import os
 import gradio as gr
-import boto3
+from werkzeug.utils import secure_filename
 
-def upload_to_s3(bucket_name, file_name, file_data):
-    s3 = boto3.resource('s3')
-    bucket = s3.Bucket(bucket_name)
-    bucket.put_object(Key=file_name, Body=file_data)
+UPLOAD_FOLDER = './'
+ALLOWED_EXTENSIONS = {'txt', 'pdf', 'doc', 'docx', 'png', 'jpg', 'jpeg', 'gif'}
 
-def file_upload(workspace_name, document_name, document_file):
-    # Assume the user is authenticated and has access to workspace_name
-    # Store the document_file in the cloud storage solution for workspace_name
-    bucket_name = f"{workspace_name}-bucket"
-    upload_to_s3(bucket_name, document_name, document_file.read())
-    return f"Document {document_name} uploaded successfully to workspace {workspace_name}."
+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,
+)
 
-workspace_input = gr.inputs.Textbox(label="Workspace Name")
-document_name_input = gr.inputs.Textbox(label="Document Name")
-document_file_input = gr.inputs.File(label="Document File")
+def allowed_file(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-def upload():
-    # Check that user is authenticated and has access to workspace_input
-    workspace_name = workspace_input.value
-    document_name = document_name_input.value
-    document_file = document_file_input.value
-    return file_upload(workspace_name, document_name, document_file)
+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 = gr.Interface(upload, inputs=[workspace_input, document_name_input, document_file_input], outputs="text")
-app.launch(debug = False)
+@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()