Testys commited on
Commit
b4a98a2
1 Parent(s): 4ee0abb

Upload 4 files

Browse files
Files changed (3) hide show
  1. .gitignore +165 -0
  2. main.py +22 -75
  3. requirements.txt +15 -5
.gitignore ADDED
@@ -0,0 +1,165 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # poetry
98
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102
+ #poetry.lock
103
+
104
+ # pdm
105
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106
+ #pdm.lock
107
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108
+ # in version control.
109
+ # https://pdm.fming.dev/#use-with-ide
110
+ .pdm.toml
111
+
112
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113
+ __pypackages__/
114
+
115
+ # Celery stuff
116
+ celerybeat-schedule
117
+ celerybeat.pid
118
+
119
+ # SageMath parsed files
120
+ *.sage.py
121
+
122
+ # Environments
123
+ .env
124
+ .venv
125
+ env/
126
+ venv/
127
+ ENV/
128
+ env.bak/
129
+ venv.bak/
130
+
131
+ # Spyder project settings
132
+ .spyderproject
133
+ .spyproject
134
+
135
+ # Rope project settings
136
+ .ropeproject
137
+
138
+ # mkdocs documentation
139
+ /site
140
+
141
+ # mypy
142
+ .mypy_cache/
143
+ .dmypy.json
144
+ dmypy.json
145
+
146
+ # Pyre type checker
147
+ .pyre/
148
+
149
+ # pytype static type analyzer
150
+ .pytype/
151
+
152
+ # Cython debug symbols
153
+ cython_debug/
154
+
155
+ # Database for Project
156
+ /data/
157
+ /images/
158
+
159
+ # PyCharm
160
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
161
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
162
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
163
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
164
+ #.idea/
165
+
main.py CHANGED
@@ -1,89 +1,36 @@
1
- import os
2
- from fastapi import FastAPI, UploadFile
3
- from fastapi.responses import JSONResponse
4
- from fastapi.param_functions import File
5
- from fastapi.middleware.cors import CORSMiddleware
6
  from typing import List
7
- import io
8
- from facenet_pytorch import MTCNN, InceptionResnetV1
9
- import torch
10
- from PIL import Image
11
-
12
- app = FastAPI()
13
-
14
- app.add_middleware(
15
- CORSMiddleware,
16
- allow_origins=["*"],
17
- allow_credentials=True,
18
- allow_methods=["*"],
19
- allow_headers=["*"],
20
  )
21
-
22
- # Set the cache directory to a writable location
23
- os.environ['TORCH_HOME'] = '/tmp/.cache/torch'
24
-
25
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
26
-
27
- mtcnn = MTCNN(keep_all=True, device=device)
28
- resnet = InceptionResnetV1(pretrained='vggface2').eval().to(device)
29
 
30
 
31
  @app.get("/", tags=["Home"])
32
  def read_root():
33
- return {"message": "Welcome to the face embeddings API!"}
 
34
 
35
  @app.get("/health", tags=["Health"])
36
  def health_check():
37
  return {"status": "ok"}
38
 
39
- @app.post("/extract", tags=["Extract Embeddings"])
40
- async def extract_embeddings(file: UploadFile = File(...)):
41
- # Load the image
42
- contents = await file.read()
43
- image = Image.open(io.BytesIO(contents)).convert('RGB')
44
-
45
- # Detect faces
46
- faces = mtcnn(image)
47
-
48
- # Check if any faces were detected
49
- if faces is None:
50
- return JSONResponse(content={"error": "No faces detected in the image"})
51
-
52
- # If faces is a list, take the first face. If it's a tensor, it's already the first (and only) face
53
- if isinstance(faces, list):
54
- face = faces[0]
55
- else:
56
- face = faces
57
-
58
- # Ensure the face tensor is 4D (batch_size, channels, height, width)
59
- if face.dim() == 3:
60
- face = face.unsqueeze(0)
61
-
62
- # Extract the face embeddings
63
- with torch.no_grad():
64
- embeddings = resnet(face).cpu().numpy().tolist()
65
-
66
- return JSONResponse(content={"embeddings": embeddings})
67
-
68
-
69
- # @app.post("/extract")
70
- # async def extract_embeddings(file: UploadFile = File(...)):
71
- # # Load the image
72
- # contents = await file.read()
73
- # image = face_recognition.load_image_file(io.BytesIO(contents))
74
-
75
- # # Find all the faces in the image
76
- # face_locations = face_recognition.face_locations(image)
77
 
78
- # # Initialize an empty list to store the face embeddings
79
- # embeddings = []
80
 
81
- # # Loop through each face location
82
- # for face_location in face_locations:
83
- # # Extract the face encoding
84
- # face_encoding = face_recognition.face_encodings(image, [face_location])[0]
85
-
86
- # # Append the face encoding to the embeddings list
87
- # embeddings.append(face_encoding.tolist())
88
 
89
- # return JSONResponse(content={"embeddings": embeddings})
 
1
+ # main.py
2
+ from fastapi import FastAPI
 
 
 
3
  from typing import List
4
+ from users.routes import router as users_router
5
+ from auth.route import router as auth_router
6
+ from orders.routes import order_router, meal_router
7
+
8
+ app = FastAPI(
9
+ title="SnapFeast API",
10
+ description="SnapFeast API for managing users, meals, and orders.",
11
+ version="1.0.0",
12
+ docs_url="/docs",
13
+ redoc_url="/redoc",
14
+ openapi_url="/openapi.json",
 
 
15
  )
16
+ app.include_router(users_router)
17
+ app.include_router(auth_router)
18
+ app.include_router(meal_router)
19
+ app.include_router(order_router)
 
 
 
 
20
 
21
 
22
  @app.get("/", tags=["Home"])
23
  def read_root():
24
+ return {"message": "Welcome to SnapFeast API!"}
25
+
26
 
27
  @app.get("/health", tags=["Health"])
28
  def health_check():
29
  return {"status": "ok"}
30
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
 
32
+ if __name__ == "__main__":
33
+ import uvicorn
34
 
35
+ uvicorn.run(app, host="0.0.0.0", port=8000, log_level="info")
 
 
 
 
 
 
36
 
 
requirements.txt CHANGED
@@ -1,9 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
1
  facenet_pytorch
2
  torch
3
- torchvision
4
  numpy
 
5
  pandas
6
- Pillow==8.2.0
7
- fastapi
8
- uvicorn
9
- python-multipart
 
1
+ # requirements.txt
2
+ fastapi
3
+ uvicorn
4
+ sqlalchemy
5
+ pydantic
6
+ pydantic[email]
7
+ pydantic-settings
8
+ passlib
9
+ python-jose[cryptography]
10
+ python-multipart
11
+ opencv-python
12
  facenet_pytorch
13
  torch
 
14
  numpy
15
+ scikit-learn
16
  pandas
17
+ surprise
18
+ joblib
19
+ psycopg2-binary