Update app.py

app.py

@@ -227,7 +227,8 @@ async def list_files():
     """List all uploaded files"""
     try:
         files = os.listdir(UPLOAD_DIR)
-        return {"files": files}
+        filtered_files = [file for file in files if not file.startswith('temp')]
+        return {"files": filtered_files}
     except Exception as e:
         logger.error(f"Error listing files: {str(e)}")
         raise HTTPException(
@@ -352,7 +353,7 @@ def convert_binary_to_image(binary_path: str, output_path: str, width: int):
 
 @app.get("/image/{file_name}")
 async def image(file_name: str):
-    if os.path.exists(file_name):
+    if os.path.exists(os.path.join(UPLOAD_DIR, file_name)):
         return FileResponse(file_name, media_type="image/png")
     return JSONResponse(content={"error": "Image not found"})
 
@@ -377,7 +378,7 @@ async def analyse_bin(file_name: str):
         result = predict_malware(img_array)
         return {
             **result,
-            "image_location": str(temp_image)
+            "image_location": str(temp_image)[8:]
         }
         
     except HTTPException as he: