org_creation / app.py
Rahul8827's picture
Update app.py
f4ca8ae
import json
import socket
import os
from datetime import datetime
import boto3
import gradio as gr
s3 = boto3.resource('s3')
def Create_Organization(org_name, org_handle):
on = org_name
o = org_handle
path = o
isExist = os.path.exists(path)
hostname = socket.gethostname()
now = datetime.now()
tim = now.strftime("%d/%m/%Y %H:%M:%S")
Organizationdetails = {}
Organizationdetails[org_handle] = {}
Organizationdetails[org_handle]['Organizationame'] = org_name
Organizationdetails[org_handle]['OrganizationHandle'] = org_handle
Organizationdetails[org_handle]['Created_by'] = hostname
Organizationdetails[org_handle]['Created_Time'] = tim
# Create a bucket in S3
bucket_name = 'myorg'
bucket = s3.Bucket(bucket_name)
# Save the organization details to S3 bucket
key = org_name + '/Organizationdetails.json'
blob = bucket.put_object(Key=key, Body=json.dumps(Organizationdetails))
return "Organization Created: " + o
def list_of_organization():
s3_client = boto3.client('s3')
bucket_name = 'myorg'
result = s3_client.list_objects(Bucket=bucket_name, Delimiter='/')
folders = [o.get('Prefix') for o in result.get('CommonPrefixes')]
return "\n".join(folders)
def main_function(action, org_name=None, org_handle=None):
if action == "Create Organization":
return Create_Organization(org_name, org_handle)
elif action == "list_of_organization":
return list_of_organization()
interface = gr.Interface(
fn=main_function,
inputs=[
gr.inputs.Dropdown(
choices=["Create Organization", "list_of_organization"],
label="Select an Action"
),
gr.inputs.Textbox(label="Organization Name"),
gr.inputs.Textbox(label="Organization Handle"),
],
outputs=gr.outputs.Textbox(label="Message"),
)
interface.launch()