Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -90,8 +90,30 @@ async def list_files():
|
|
90 |
except FileNotFoundError:
|
91 |
raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
|
92 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
93 |
|
94 |
-
@app.get("/files/{origin_id}")
|
95 |
async def get_file_by_origin_id(origin_id: int):
|
96 |
try:
|
97 |
for filename in os.listdir(BASE_DIR):
|
|
|
90 |
except FileNotFoundError:
|
91 |
raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
|
92 |
|
93 |
+
@app.get("/files/similar")
|
94 |
+
async def list_similar_files():
|
95 |
+
try:
|
96 |
+
files_data = []
|
97 |
+
for filename in os.listdir(BASE_DIR):
|
98 |
+
filepath = os.path.join(BASE_DIR, filename)
|
99 |
+
if os.path.isfile(filepath):
|
100 |
+
try:
|
101 |
+
with open(filepath, "r") as f:
|
102 |
+
file_content = f.read()
|
103 |
+
try:
|
104 |
+
file_content_json = json.loads(file_content)
|
105 |
+
# Check for similarityOrb and filter
|
106 |
+
if "similarityOrb" in file_content_json and file_content_json["similarityOrb"] > 0:
|
107 |
+
files_data.append({"filename": filename, "content": file_content_json})
|
108 |
+
except json.JSONDecodeError:
|
109 |
+
pass # Skip files that are not valid JSON
|
110 |
+
except (IOError, OSError) as e:
|
111 |
+
raise HTTPException(status_code=500, detail=f"Erro ao ler o ficheiro {filename}: {e}")
|
112 |
+
return JSONResponse({"files_data": files_data})
|
113 |
+
except FileNotFoundError:
|
114 |
+
raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
|
115 |
|
116 |
+
@app.get("/files/find/{origin_id}")
|
117 |
async def get_file_by_origin_id(origin_id: int):
|
118 |
try:
|
119 |
for filename in os.listdir(BASE_DIR):
|