priaansh commited on
Commit
0ce52a3
·
verified ·
1 Parent(s): d90b6b7

Rename config.py to app.py

Browse files
Files changed (2) hide show
  1. app.py +52 -0
  2. config.py +0 -14
app.py ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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)
config.py DELETED
@@ -1,14 +0,0 @@
1
- import os
2
-
3
- def get_env_var(name: str, required=True, default=None):
4
- value = os.getenv(name, default)
5
- if required and not value:
6
- raise EnvironmentError(f"Missing required environment variable: {name}")
7
- return value
8
-
9
- API_KEY = get_env_var("API_KEY")
10
- S3_ENDPOINT_URL = get_env_var("S3_ENDPOINT_URL")
11
- S3_ACCESS_KEY = get_env_var("S3_ACCESS_KEY")
12
- S3_SECRET_KEY = get_env_var("S3_SECRET_KEY")
13
- S3_BUCKET_NAME = get_env_var("S3_BUCKET_NAME")
14
- S3_REGION = get_env_var("S3_REGION")