priaansh commited on
Commit
ddc33cb
·
verified ·
1 Parent(s): 564d0db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -39
app.py CHANGED
@@ -1,52 +1,25 @@
1
  import os
2
- import boto3
3
  from flask import Flask, jsonify
4
 
5
- app = Flask(__name__)
6
-
7
- # Retrieve environment variables
8
- API_KEY = os.getenv("API_KEY")
9
- S3_ENDPOINT_URL = os.getenv("S3_ENDPOINT_URL")
10
- S3_ACCESS_KEY = os.getenv("S3_ACCESS_KEY")
11
- S3_SECRET_KEY = os.getenv("S3_SECRET_KEY")
12
- S3_BUCKET_NAME = os.getenv("S3_BUCKET_NAME")
13
- S3_REGION = os.getenv("S3_REGION")
14
-
15
  # Validate essential environment variables
16
- required_vars = {
17
- "API_KEY": API_KEY,
18
- "S3_ENDPOINT_URL": S3_ENDPOINT_URL,
19
- "S3_ACCESS_KEY": S3_ACCESS_KEY,
20
- "S3_SECRET_KEY": S3_SECRET_KEY,
21
- "S3_BUCKET_NAME": S3_BUCKET_NAME,
22
- "S3_REGION": S3_REGION
23
- }
24
-
25
- missing_vars = [var for var, value in required_vars.items() if not value]
26
  if missing_vars:
27
  raise EnvironmentError(f"Missing required environment variables: {', '.join(missing_vars)}")
28
 
29
- # Initialize the S3 client
30
- s3_client = boto3.client(
31
- "s3",
32
- endpoint_url=S3_ENDPOINT_URL,
33
- aws_access_key_id=S3_ACCESS_KEY,
34
- aws_secret_access_key=S3_SECRET_KEY,
35
- region_name=S3_REGION
36
- )
37
 
38
  @app.route("/")
39
  def home():
40
- return jsonify({"message": "Welcome to the No-Code Architect's Toolkit!"})
41
-
42
- @app.route("/list-files")
43
- def list_files():
44
- try:
45
- response = s3_client.list_objects_v2(Bucket=S3_BUCKET_NAME)
46
- files = [obj["Key"] for obj in response.get("Contents", [])]
47
- return jsonify({"files": files})
48
- except Exception as e:
49
- return jsonify({"error": str(e)}), 500
50
 
51
  if __name__ == "__main__":
52
  app.run(host="0.0.0.0", port=8080)
 
1
  import os
 
2
  from flask import Flask, jsonify
3
 
 
 
 
 
 
 
 
 
 
 
4
  # Validate essential environment variables
5
+ required_env_vars = [
6
+ "API_KEY",
7
+ "S3_ENDPOINT_URL",
8
+ "S3_ACCESS_KEY",
9
+ "S3_SECRET_KEY",
10
+ "S3_BUCKET_NAME",
11
+ "S3_REGION"
12
+ ]
13
+
14
+ missing_vars = [var for var in required_env_vars if not os.getenv(var)]
15
  if missing_vars:
16
  raise EnvironmentError(f"Missing required environment variables: {', '.join(missing_vars)}")
17
 
18
+ app = Flask(__name__)
 
 
 
 
 
 
 
19
 
20
  @app.route("/")
21
  def home():
22
+ return jsonify({"message": "Welcome to No-Code Architect's Toolkit"})
 
 
 
 
 
 
 
 
 
23
 
24
  if __name__ == "__main__":
25
  app.run(host="0.0.0.0", port=8080)