Update app.py

app.py

@@ -5,44 +5,47 @@ from datetime import datetime
 import boto3
 import gradio as gr
 
-# Create an S3 resource
-s3 = boto3.resource('s3')
-bucket_name = 'myorg'
-
-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_handle + '/Organizationdetails.json'
-    blob = bucket.put_object(Key=key, Body=json.dumps(Organizationdetails))
-    
-    return "Organization Created: " + o
-organizations = []
-objects = s3.list_objects(Bucket=bucket_name)['Contents']
-for object in s3.list_objects('bucket_name')['Contents']:
-    data = json.loads(s3.get_object(Key=object['Key'])['Body'].read())
-    organizations.append(data['organization_name'])
-org_names = ", ".join(organizations)
+# S3 client 
+s3 = boto3.client('s3') 
+
+# Bucket name
+bucket_name = 'your_bucket_name'
+
+def create_organization(org_name, org_handle):
+
+  # 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") 
+  }
+
+  # Save to S3
+  s3.put_object(
+    Bucket=bucket_name, 
+    Key=f"{org_handle}/details.json",
+    Body=json.dumps(org_details)
+  )
+
+  # 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)
+
 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="Existing Organizations"),
+  gr.outputs.Textbox(label="Message")
 ]
-outputs = [gr.outputs.Textbox(gr.outputs.Textbox(label="Existing Organizations"),gr.outputs.Textbox(label="Message")]
 
-gr.Interface(Create_Organization, inputs, outputs).launch()
+gr.Interface(fn=create_organization, inputs=inputs, outputs=outputs).launch()