Update app.py

app.py

@@ -5,47 +5,65 @@ from datetime import datetime
 import boto3
 import gradio as gr
 
-# S3 client 
-s3 = boto3.client('s3') 
 
-# Bucket name
-bucket_name = 'myorg'
 
-def create_organization(org_name, org_handle):
+s3 = boto3.resource('s3')
 
-  # Create organization 
-  org_details = {
-    'name': org_name,
-    'handle': org_handle,
-    'created_by': socket.gethostname(),
-    'created_at': datetime.now().strftime("%d/%m/%Y %H:%M:%S") 
-  }
+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
 
-  # Save to S3
-  s3.put_object(
-    Bucket=bucket_name, 
-    Key=f"{org_handle}/details.json",
-    Body=json.dumps(org_details)
-  )
+    # Create a bucket in S3
+    bucket_name = 'myorg'
+    bucket = s3.Bucket(bucket_name)
 
-  # Get existing organizations
-  existing_orgs = []
-  objects = s3.list_objects(Bucket=bucket_name)['Contents']
-
-  for obj in objects:
-    data = json.loads(s3.get_object(Bucket=bucket_name, Key=obj['Key'])['Body'].read())
-    existing_orgs.append(data['name'])
-
-  return ", ".join(existing_orgs)
+    # 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
 
 inputs = [
-  gr.inputs.Textbox(label="Organization Name"),
-  gr.inputs.Textbox(label="Organization Handle") 
+    gr.inputs.Textbox(label="Organization Name"),
+    gr.inputs.Textbox(label="Organization Handle"),
 ]
+outputs = gr.outputs.Textbox(label="Message")
+
+gr.Interface(Create_Organization, inputs, outputs).launch()
+
+
+import gradio as gr
+
+def list_s3_folders(bucket_name):
+    import boto3
+    
+    s3 = boto3.client('s3')
+    result = s3.list_objects(Bucket=bucket_name, Delimiter='/')
+    
+    folders = [o.get('Prefix') for o in result.get('CommonPrefixes')]
+    return folders
+
+iface = gr.Interface(
+    fn=list_s3_folders,
+    inputs="text",
+    outputs="text",
+    layout="vertical",
+    title="List S3 Folders",
+    description="Enter the S3 bucket name to list all folders in the bucket.",
+    examples=[["myorg"]]
+)
+
+iface.launch()
 
-outputs = [
-  gr.outputs.Textbox(label="Existing Organizations"),
-  gr.outputs.Textbox(label="Message")
-]
 
-gr.Interface(fn=create_organization, inputs=inputs, outputs=outputs).launch()